Pass session to TrainerStrategy

GitOrigin-RevId: f1a6ebd6ef3e64e081db8c38ac7f0b2aeba8b6ab
gretelai · Feb 22, 2024 · 5da11ca · 5da11ca
1 parent 79b5aa0
commit 5da11ca
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/gretel_trainer/benchmark/session.py b/src/gretel_trainer/benchmark/session.py
@@ -264,6 +264,7 @@ def _create_gretel_strategy(
                 run_identifier=run_identifier,
                 project_name=trainer_project_name,
                 config=self._config,
+                session=self._session,
             )
         else:
             return GretelSDKStrategy(