summaryrefslogtreecommitdiff
path: root/training/artifacts.py
diff options
context:
space:
mode:
Diffstat (limited to 'training/artifacts.py')
-rw-r--r--training/artifacts.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/training/artifacts.py b/training/artifacts.py
index 2cb8bda..f7af085 100644
--- a/training/artifacts.py
+++ b/training/artifacts.py
@@ -1,14 +1,15 @@
"""Fetches model artifacts from wandb."""
-from datetime import datetime
-from pathlib import Path
import shutil
import sys
+from datetime import datetime
+from pathlib import Path
from typing import Optional
import click
from loguru import logger as log
-from training import metadata
+
import wandb
+from training import metadata
from wandb.apis.public import Run