Skip to content

Commit 11080dc

Browse files
Remove AsyncTrajectory group
1 parent aad0b46 commit 11080dc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/art/cli.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from . import dev
1313
from .local import LocalBackend
1414
from .model import Model, TrainableModel, list_trash, restore_from_trash, empty_trash, delete_model
15-
from .trajectories import TrajectoryGroup, AsyncTrajectoryGroup
15+
from .trajectories import TrajectoryGroup
1616
from .types import TrainConfig
1717
from .utils.deploy_model import LoRADeploymentProvider
1818
from .errors import ARTError
@@ -69,7 +69,7 @@ async def _log(
6969
@app.post("/_train_model")
7070
async def _train_model(
7171
model: TrainableModel,
72-
trajectory_groups: list[AsyncTrajectoryGroup],
72+
trajectory_groups: list[TrajectoryGroup],
7373
config: TrainConfig,
7474
dev_config: dev.TrainConfig,
7575
verbose: bool = Body(False),
@@ -173,4 +173,4 @@ def restore_model(project_name: str, model_name: str):
173173
def empty_trashed_models(project_name: str):
174174
"""Permanently delete all models in the trash."""
175175
empty_trash(project_name)
176-
print("Trash has been emptied.")
176+
print("Trash has been emptied.")

0 commit comments

Comments
 (0)