Skip to content

Commit

Permalink
Merge artifact reader with writer
Browse files Browse the repository at this point in the history
  • Loading branch information
kstathou committed Dec 4, 2023
1 parent 039a58e commit e6d214c
Showing 1 changed file with 2 additions and 17 deletions.
19 changes: 2 additions & 17 deletions src/llm_stack/wandb_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,16 +59,8 @@ def id(self) -> str:
return self.run.id


class ArtifactWriter(WandbRun):
"""Write an artifact to wandb.
Types of artifacts:
- Data (raw data, training, validation, test, metrics)
- ML Model checkpoints
For now, it's a glorified wrapper of the wandb API.
"""
class ArtifactHandler(WandbRun):
"""Read and write artifacts stored in wandb."""

def __init__(self, project: str, job_type: str, **kwargs) -> None:
super().__init__(project, job_type, **kwargs)
Expand Down Expand Up @@ -128,13 +120,6 @@ def _log_artifact(

self.run.log_artifact(artifact)


class ArtifactReader(WandbRun):
"""Read an artifact that has been stored wandb."""

def __init__(self, project: str, job_type: str, **kwargs) -> None:
super().__init__(project, job_type, **kwargs)

def read_artifact(
self,
name: str,
Expand Down

0 comments on commit e6d214c

Please sign in to comment.