gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-28 07:57:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Add framework handling for task coding failure. (#176)

Browse Source 
* Add framework handling for task coding failure.

* fix a ci bug
This commit is contained in:
WinstonLiyt
2024-08-07 19:01:46 +08:00
committed by GitHub
parent 473a4e9c02
commit 1280e681bb
4 changed files with 28 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/factor_coder/factor.py
+11
View File
@@ -26,12 +26,14 @@ class FactorTask(Task):
factor_formulation,
variables: dict = {},
resource: str = None,
factor_implementation: bool = False,
) -> None:
self.factor_name = factor_name
self.factor_description = factor_description
self.factor_formulation = factor_formulation
self.variables = variables
self.factor_resources = resource
self.factor_implementation = factor_implementation
def get_task_information(self):
return f"""factor_name: {self.factor_name}
@@ -39,6 +41,15 @@ factor_description: {self.factor_description}
factor_formulation: {self.factor_formulation}
variables: {str(self.variables)}"""
def get_task_information_and_implementation_result(self):
return {
"factor_name": self.factor_name,
"factor_description": self.factor_description,
"factor_formulation": self.factor_formulation,
"variables": str(self.variables),
"factor_implementation": str(self.factor_implementation),
}
@staticmethod
def from_dict(dict):
return FactorTask(**dict)