fix paths

Author: pattonw

pyproject.toml

@@ -32,3 +32,6 @@ predict = "volara-torch.blockwise.predict:Predict"
 
 [tool.uv.sources]
 volara = { git = "https://github.com/e11bio/volara" }
+
+[tool.ruff]
+lint.select = ["F", "W", "I001"]

src/volara-torch/blockwise/predict.py

@@ -11,11 +11,11 @@
 from funlib.persistence import Array
 from gunpowder import ArrayKey, Batch, BatchProvider
 from pydantic import Field
+from volara.blockwise import BlockwiseTask
+from volara.datasets import LSD, Affs, Dataset, Raw
+from volara.utils import PydanticCoordinate
 
-from ..datasets import LSD, Affs, Dataset, Raw
 from ..models import Model, TorchModel
-from ..utils import PydanticCoordinate
-from .blockwise import BlockwiseTask
 
 logger = logging.getLogger(__file__)
 

tests/test_models.py

@@ -1,8 +1,7 @@
 import numpy as np
 import pytest
 import torch
-
-from volara.models import TorchModel
+from volara_torch.models import TorchModel
 
 
 @pytest.mark.parametrize("save_type", ["jit", "pickle"])