Skip to content

Commit 8135f22

Browse files
committed
linting
1 parent c7b7750 commit 8135f22

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

scripts/migrate_old_model.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
"""Script to migrate old pvnet-summation models (v0.3.7) which are hosted on huggingface to current
2-
version"""
2+
"""
3+
34
import datetime
45
import os
5-
from importlib.metadata import version
66
import tempfile
7+
from importlib.metadata import version
78

89
import torch
910
import yaml
@@ -12,7 +13,10 @@
1213

1314
from pvnet_summation.models import BaseModel
1415
from pvnet_summation.utils import (
15-
MODEL_CARD_NAME, MODEL_CONFIG_NAME, PYTORCH_WEIGHTS_NAME, DATAMODULE_CONFIG_NAME,
16+
DATAMODULE_CONFIG_NAME,
17+
MODEL_CARD_NAME,
18+
MODEL_CONFIG_NAME,
19+
PYTORCH_WEIGHTS_NAME,
1620
)
1721

1822
# ------------------------------------------

0 commit comments

Comments
 (0)