Fixing imports

deepsensor/data/task.py

@@ -6,8 +6,8 @@
 import plum
 import copy
 
-from deepsensor import backend
-from deepsensor.errors import TaskSetIndexError, GriddedDataError
+from . import backend
+from ..errors import TaskSetIndexError, GriddedDataError
 
 
 class Task(dict):

deepsensor/model/convnp.py

@@ -1,33 +1,31 @@
 import copy
 import os.path
 import json
-from typing import Union, List, Literal, Tuple, Optional, types
+from typing import Union, List, Literal, Optional
 import warnings
-from typing import Union, List
 
 import lab as B
 import numpy as np
 import warnings
 from matrix import Diagonal
 from plum import ModuleType, dispatch
 
-import deepsensor.data.task
-from deepsensor import backend
-from deepsensor.data.loader import TaskLoader
-from deepsensor.data.processor import DataProcessor
-from deepsensor.data.task import (
+from .. import backend
+from ..data.loader import TaskLoader
+from ..data.processor import DataProcessor
+from ..data.task import (
     Task,
     flatten_gridded_data_in_task,
     flatten_X,
     flatten_Y,
 )
-from deepsensor.model.defaults import (
+from ..model.defaults import (
     gen_ppu,
     gen_encoder_scales,
     gen_decoder_scale,
 )
-from deepsensor.model.model import DeepSensorModel
-from deepsensor.model.nps import (
+from ..model.model import DeepSensorModel
+from ..model.nps import (
     construct_neural_process,
     convert_task_to_nps_args,
     run_nps_model,

deepsensor/model/nps.py

@@ -1,4 +1,4 @@
-from deepsensor import backend
+from .. import backend
 import lab as B
 
 from deepsensor.data.task import Task

deepsensor/tensorflow/__init__.py

@@ -20,8 +20,8 @@ def convert_to_tensor(arr):
     return tf.convert_to_tensor(arr)
 
 
-from deepsensor import config as deepsensor_config
-from deepsensor import backend
+from .. import config as deepsensor_config
+from .. import backend
 
 backend.nps = nps
 backend.model = tf.keras.Model

deepsensor/torch/__init__.py

@@ -20,8 +20,8 @@ def convert_to_tensor(arr):
     return torch.tensor(arr)
 
 
-from deepsensor import config as deepsensor_config
-from deepsensor import backend
+from .. import config as deepsensor_config
+from .. import backend
 
 backend.nps = nps
 backend.model = torch.nn.Module