lobster beignet dependencies

Genentech · May 14, 2024 · 2991399 · 2991399
1 parent dcd901a
commit 2991399
Show file tree

Hide file tree

Showing 8 changed files with 12 additions and 112 deletions.
diff --git a/src/beignet/features/__init__.py b/src/beignet/features/__init__.py
@@ -1,5 +1,5 @@
-from ._feature import Feature
+from ._features import Feature
 
 __all__ = [
     "Feature",
-]
+]
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/beignet/__init__.py b/tests/beignet/__init__.py
diff --git a/tests/beignet/features/conftest.py b/tests/beignet/features/conftest.py
@@ -0,0 +1,8 @@
+import pytest
+import torch
+from yeji.features._feature import Feature
+
+
+@pytest.fixture
+def feature() -> Feature:
+    return Feature(torch.tensor([1, 2, 3]))
diff --git a/tests/beignet/features/test__feature.py b/tests/beignet/features/test__feature.py
@@ -24,8 +24,6 @@ def test___torch_function__(self, feature: Feature):
             (feature, torch.tensor([1, 2, 3])),
         )
 
-        assert torch.is_tensor(result)
-
         assert not isinstance(result, Feature)
 
     def test_device(self, feature: Feature):

diff --git a/tests/beignet/io/test__load_omega_config.py b/tests/beignet/io/test__load_omega_config.py
diff --git a/tests/beignet/io/test__load_parquet.py b/tests/beignet/io/test__load_parquet.py
@@ -1,6 +1,6 @@
 import pandas as pd
 import pytest
-from prescient.io._load_parquet import load_parquet_to_pandas
+from beignet.io._load_parquet import load_parquet_to_pandas
 
 
 @pytest.fixture

diff --git a/tests/beignet/io/test__parse_s3_path.py b/tests/beignet/io/test__parse_s3_path.py
@@ -1,4 +1,4 @@
-from prescient.io._parse_s3_path import parse_s3_path
+from beignet.io._parse_s3_path import parse_s3_path
 
 
 def test_parse_s3_path():