inital ai hub implementation
This commit is contained in:
16
src/state.py
16
src/state.py
@@ -33,6 +33,22 @@ class CliStateStore:
|
||||
value = self.get("last_training_job")
|
||||
return str(value) if value else None
|
||||
|
||||
def get_last_model_artifact(self) -> str | None:
|
||||
value = self.get("last_model_artifact")
|
||||
return str(value) if value else None
|
||||
|
||||
def get_last_quantized_model_id(self) -> str | None:
|
||||
value = self.get("last_quantized_model_id")
|
||||
return str(value) if value else None
|
||||
|
||||
def get_last_compiled_model_id(self) -> str | None:
|
||||
value = self.get("last_compiled_model_id")
|
||||
return str(value) if value else None
|
||||
|
||||
def get_last_downloaded_model(self) -> str | None:
|
||||
value = self.get("last_downloaded_model")
|
||||
return str(value) if value else None
|
||||
|
||||
def set_last_training_job(self, job_name: str) -> None:
|
||||
self.update(last_training_job=job_name)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user