refactor: util routine

wcxve · Jan 31, 2024 · 94a29de · 94a29de
1 parent d9c5cfd
commit 94a29de
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/src/elisa/__init__.py b/src/elisa/__init__.py
@@ -1,8 +1,5 @@
-from .data import *
-from .model import *
-from .infer import *
-from .util import *
-from .__about__ import __version__
+from .model import *  # noqa: F403
+from .util import jax_enable_x64, set_cpu_cores
 
 jax_enable_x64(True)
 set_cpu_cores(4)
diff --git a/src/elisa/util/__init__.py b/src/elisa/util/__init__.py
@@ -0,0 +1 @@
+from .config import *  # noqa: F403