Merge pull request #161 from AllenNeuralDynamics:fix-deserialization

Fix issue with task not being able to be deserialized from dictionary in certain edge cases

Author: bruno-f-cruz

pyproject.toml

@@ -11,7 +11,7 @@ authors = [
 license = "MIT"
 license-files = ["LICENSE"]
 requires-python = ">=3.11"
-version = "0.8.0"
+version = "0.8.1rc0"
 readme = {file = "README.md", content-type = "text/markdown"}
 
 classifiers = [

src/clabe/pickers/dataverse.py

@@ -578,7 +578,10 @@ def pick_trainer_state(self, task_logic_model: Type[TTaskLogic]) -> tuple[Traine
         assert self._trainer_state is not None
         if not self._trainer_state.is_on_curriculum:
             logging.warning("Deserialized TrainerState is NOT on curriculum.")
-        return (self.trainer_state, task_logic_model.model_validate(self.trainer_state.stage.task.model_dump()))
+        return (
+            self.trainer_state,
+            task_logic_model.model_validate_json(self.trainer_state.stage.task.model_dump_json()),
+        )
 
     def push_new_suggestion(self, trainer_state: TrainerState) -> None:
         """

src/clabe/pickers/default_behavior.py

@@ -342,7 +342,10 @@ def pick_trainer_state(self, task_logic_model: Type[TTaskLogic]) -> tuple[Traine
             logging.warning("Deserialized TrainerState is NOT on curriculum.")
 
         assert self._trainer_state.stage is not None
-        return (self.trainer_state, task_logic_model.model_validate(self.trainer_state.stage.task.model_dump()))
+        return (
+            self.trainer_state,
+            task_logic_model.model_validate_json(self.trainer_state.stage.task.model_dump_json()),
+        )
 
     def choose_subject(self, directory: str | os.PathLike) -> str:
         """