Skip to content

Commit d0a88ce

Browse files
author
nik
committed
Fix redis import
1 parent 31209b2 commit d0a88ce

File tree

4 files changed

+4
-6
lines changed

4 files changed

+4
-6
lines changed

label_studio_ml/__init__.py

-2
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
from .model import LabelStudioMLBase
2-
31
# Package name
42
package_name = 'label-studio-ml'
53

label_studio_ml/examples/dummy_model.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import random
22

3-
from label_studio_ml import LabelStudioMLBase
3+
from label_studio_ml.model import LabelStudioMLBase
44

55

66
class DummyModel(LabelStudioMLBase):

label_studio_ml/examples/pytorch_transfer_learning.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@
1313
from torch.utils.data import Dataset, DataLoader
1414
from torchvision import models, transforms
1515

16-
from label_studio.ml import LabelStudioMLBase
17-
from label_studio.ml.utils import get_single_tag_keys, get_choice, is_skipped
16+
from label_studio_ml.model import LabelStudioMLBase
17+
from label_studio_ml.utils import get_single_tag_keys, get_choice, is_skipped
1818

1919

2020
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
'Operating System :: OS Independent',
2929
],
3030
install_requires=requirements,
31-
python_requires='>=3.6, <3.9',
31+
python_requires='>=3.6',
3232
entry_points={
3333
'console_scripts': [
3434
'label-studio-ml=label_studio_ml.server:main'

0 commit comments

Comments
 (0)