Adding _get_state, _load_state to Experiment class

hegelai · Oct 31, 2023 · 5a102ab · 5a102ab
1 parent c60d32e
commit 5a102ab
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/prompttools/experiment/experiments/experiment.py b/prompttools/experiment/experiments/experiment.py
@@ -641,3 +641,10 @@ def _get_model_names(self):
 
  def _get_prompts(self):
  pass
+
+ def _get_state(self):
+ raise NotImplementedError("Should be implemented by specific harness class.")
+
+ @classmethod
+ def _load_state(cls, state, experiment_id: str, revision_id: str, experiment_type_str: str):
+ raise NotImplementedError("Should be implemented by specific harness class.")