Adding parameters to self_kwargs (#338)

truera · Aug 1, 2023 · d2d61bb · d2d61bb
1 parent 5cf0cef
commit d2d61bb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/trulens_eval/trulens_eval/feedback.py b/trulens_eval/trulens_eval/feedback.py
@@ -1043,6 +1043,7 @@ def __init__(
 
         # TODO: why was self_kwargs required here independently of kwargs?
         self_kwargs = dict()
+        self_kwargs.update(**kwargs)
         self_kwargs['model_engine'] = model_engine
         self_kwargs['endpoint'] = OpenAIEndpoint(*args, **kwargs)