Fix imports

coudertlab · Apr 12, 2024 · a3a05cb · a3a05cb
1 parent f7c100b
commit a3a05cb
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/src/ELATE/__init__.py b/src/ELATE/__init__.py
@@ -0,0 +1,4 @@
+# -*- coding: utf-8 -*-
+
+from .elastic import Elastic2D, Elastic, ElasticOrtho
+from .elate import ELATE, ELATE_MaterialsProject, plot3D, wait3D
diff --git a/src/ELATE/elate.py b/src/ELATE/elate.py
@@ -16,7 +16,7 @@
 import numpy as np
 from scipy import optimize
 
-import elastic
+from ELATE import elastic
 
 __author__ = "Romain Gaillac and François-Xavier Coudert"
 __version__ = "2024.03.15"

diff --git a/tests/test_init_1.py b/tests/test_init_1.py
@@ -2,5 +2,5 @@
 
 
 def test_init_1():
-    import elastic
+    import ELATE