betti1 and chi estimator will use numba as default

sambit-giri · Dec 30, 2024 · 4ce4dc1 · 4ce4dc1
1 parent e1d3b2f
commit 4ce4dc1
Show file tree

Hide file tree

Showing 6 changed files with 165 additions and 160 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
 
 [tool.poetry]
 name = "tools21cm"
-version = "2.2.7"
+version = "2.2.8"
 description = "A package providing tools to analyse cosmological simulations of reionization"
 authors = ["Sambit Giri <sambit.giri@gmail.com>"]
 license = "MIT"
@@ -32,6 +32,8 @@ tqdm = "^4.60"
 joblib = "^1.0"
 pandas = "^1.3"
 pytest = "^6.2"
+torch = "^2.1.0"
+numba = "^0.57.1"
 
 [tool.poetry.dev-dependencies]
 pytest = "^6.2"  # If you want pytest to be part of development dependencies, but you already have it above too.

diff --git a/requirements.txt b/requirements.txt
@@ -9,3 +9,5 @@ tqdm
 joblib
 pandas
 pytest
+torch==2.1.0
+numba
diff --git a/setup.py b/setup.py
@@ -36,7 +36,7 @@
 
 setup(
     name='tools21cm',
-    version='2.2.7',
+    version='2.2.8',
     author='Sambit Giri',
     author_email='sambit.giri@gmail.com',
     packages=find_packages(where="src"),
-Original file line number
+Diff line change
@@ Expand Up / @@ -9,3 +9,5 @@ tqdm @@
     joblib
     pandas
     pytest
+    torch==2.1.0
+    numba