major bug in pip installation

haruishi43 · Dec 22, 2020 · 23dc81a · 23dc81a
1 parent 803249a
commit 23dc81a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/equilib/__init__.py b/equilib/__init__.py
@@ -1,9 +1,9 @@
 #!/usr/bin/env python3
 
-from .cube2equi.base import Cube2Equi, cube2equi
-from .equi2cube.base import Equi2Cube, equi2cube
-from .equi2equi.base import Equi2Equi, equi2equi
-from .equi2pers.base import Equi2Pers, equi2pers
+from equilib.cube2equi.base import Cube2Equi, cube2equi
+from equilib.equi2cube.base import Equi2Cube, equi2cube
+from equilib.equi2equi.base import Equi2Equi, equi2equi
+from equilib.equi2pers.base import Equi2Pers, equi2pers
 
 __all__ = [
     "Cube2Equi",

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,3 +1,11 @@
+[build-system]
+requires = [
+  "numpy",
+  "setuptools",
+  "torch",
+  "torchvision",
+  "wheel",
+]
 [tool.black]
 line_length = 80
 exclude = '''