gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-03 10:27:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix json load error (#386)

Browse Source 
* fix json load error

* fix os.symlink error

* reformat with black

* fix os.symlink error

* reformat with isort
This commit is contained in:
Linlang
2024-10-10 11:58:15 +08:00
committed by GitHub
parent 9f6b457aa5
commit 5402c00165
2 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/core/experiment.py
+5 -1
View File
@@ -1,6 +1,7 @@
from __future__ import annotations
import os
import platform
import shutil
import uuid
from abc import ABC, abstractmethod
@@ -122,7 +123,10 @@ class FBWorkspace(Workspace):
workspace_data_file_path = workspace_path / data_file_path.name
if workspace_data_file_path.exists():
workspace_data_file_path.unlink()
os.symlink(data_file_path, workspace_data_file_path)
if platform.system() == "Linux":
os.symlink(data_file_path, workspace_data_file_path)
if platform.system() == "Windows":
os.link(data_file_path, workspace_data_file_path)
def inject_code(self, **files: str) -> None:
"""
rdagent/utils/env.py
+9 -6
View File
@@ -208,12 +208,15 @@ class DockerEnv(Env[DockerConf]):
with Progress(SpinnerColumn(), TextColumn("{task.description}")) as p:
task = p.add_task("[cyan]Building image...")
for part in resp_stream:
status_dict = json.loads(part)
if "error" in status_dict:
p.update(task, description=f"[red]error: {status_dict['error']}")
raise docker.errors.BuildError(status_dict["error"], "")
if "stream" in status_dict:
p.update(task, description=status_dict["stream"])
lines = part.decode("utf-8").split("\r\n")
for line in lines:
if line.strip():
status_dict = json.loads(line)
if "error" in status_dict:
p.update(task, description=f"[red]error: {status_dict['error']}")
raise docker.errors.BuildError(status_dict["error"], "")
if "stream" in status_dict:
p.update(task, description=status_dict["stream"])
logger.info(f"Finished building the image from dockerfile: {self.conf.dockerfile_folder_path}")
try:
client.images.get(self.conf.image)