From 5a102abff7ebc6da5f509d1fa8f93eac89354cf2 Mon Sep 17 00:00:00 2001 From: Kevin Date: Tue, 31 Oct 2023 00:40:04 -0400 Subject: [PATCH] Adding _get_state, _load_state to Experiment class --- prompttools/experiment/experiments/experiment.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/prompttools/experiment/experiments/experiment.py b/prompttools/experiment/experiments/experiment.py index 7e7393f..41ee2d5 100644 --- 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.")