Merge pull request #136 from aiverify-foundation/ms-818

fix bug on model rebuilding
aiverify-foundation · Dec 6, 2024 · 9bc4eb8 · 9bc4eb8
2 parents 05443e6 + 825b74e
commit 9bc4eb8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/runners-modules/benchmarking.py b/runners-modules/benchmarking.py
@@ -1002,3 +1002,4 @@ def from_tuple(cls, cache_record: tuple) -> PromptArguments:
             random_seed=cache_record[12],
             system_prompt=cache_record[13],
         )
+PromptArguments.model_rebuild()
diff --git a/runners-modules/redteaming.py b/runners-modules/redteaming.py
@@ -417,3 +417,4 @@ def to_dict(self) -> dict:
             "duration": str(self.connector_prompt.duration),
             "start_time": self.start_time,
         }
+RedTeamingPromptArguments.model_rebuild()