From 94a29de5793fe009fde5cd63e74f7467fde97cde Mon Sep 17 00:00:00 2001 From: xuewc Date: Thu, 1 Feb 2024 06:56:59 +0800 Subject: [PATCH] refactor: util routine --- src/elisa/__init__.py | 7 ++----- src/elisa/util/__init__.py | 1 + 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/src/elisa/__init__.py b/src/elisa/__init__.py index a7ee4046..4899f604 100644 --- 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 index e69de29b..907ef6c8 100644 --- a/src/elisa/util/__init__.py +++ b/src/elisa/util/__init__.py @@ -0,0 +1 @@ +from .config import * # noqa: F403