Skip to content

Commit 2335282

Browse files
refactor: move model trash to top-level .art/trash/<project>/models
1 parent eec768f commit 2335282

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/art/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from .openai import patch_openai
1111
from .trajectories import Trajectory, TrajectoryGroup
1212
from .types import TrainConfig
13-
from .utils.output_dirs import get_models_dir, get_model_dir
13+
from .utils.output_dirs import get_default_art_path, get_models_dir, get_model_dir
1414

1515
if TYPE_CHECKING:
1616
from art.backend import Backend
@@ -364,8 +364,8 @@ async def train(
364364

365365

366366
def _get_trash_dir(project_name: str) -> str:
367-
models_dir = get_models_dir(project_name)
368-
return os.path.join(models_dir, ".trash")
367+
art_path = get_default_art_path()
368+
return os.path.join(art_path, "trash", project_name, "models")
369369

370370

371371
def move_to_trash(project_name: str, model_name: str):

0 commit comments

Comments
 (0)