fix: Imports in predictions

antwxne · Jun 9, 2022 · 4384db9 · 4384db9
1 parent 5af4a06
commit 4384db9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="CVAT wrapper",
-    version="0.0.8",
+    version="0.0.9",
     author="antwxne",
     author_email="[email protected]",
     description="Python wrapper for CVAT API",

diff --git a/src/CVAT/Predictions/Foodvisor.py b/src/CVAT/Predictions/Foodvisor.py
@@ -2,7 +2,7 @@
 # Created by [email protected] at 5/18/22
 
 from .Interface import IPrediction
-from src.CVAT.data_types import LabeledData, Image, PatchedLabel
+from ..data_types import LabeledData, Image, PatchedLabel
 
 
 class Foodvisor(IPrediction):

diff --git a/src/CVAT/Predictions/Interface.py b/src/CVAT/Predictions/Interface.py
@@ -2,7 +2,7 @@
 import abc
 from typing import Union
 
-from src.CVAT.data_types import LabeledData
+from ..data_types import LabeledData
 
 
 # This class is an interface for a prediction object.