summaryrefslogtreecommitdiff
path: root/training/artifacts.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2024-04-15 21:47:54 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2024-04-15 21:47:54 +0200
commit9ce21f569ecac03f15f2ad669fde3dd4a512f8cc (patch)
treee6f5bbf4cfe758788fd6ad3679b714d4ecfad568 /training/artifacts.py
parenta93dcc5b9c8160a441c5b5f99f2f59264778ef91 (diff)
Format
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