Merge pull request #4 from ASEM000/v0.2

V0.1.3
ASEM000 · Mar 2, 2023 · 1fb2c24 · 1fb2c24
2 parents 2719c7e + e200ddf
commit 1fb2c24
Show file tree

Hide file tree

Showing 19 changed files with 547 additions and 579 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -10,7 +10,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest]
-        python-version: ["3.7", "3.8", "3.9", "3.10"]
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
 
     steps:
       - uses: actions/checkout@v2
@@ -21,7 +21,8 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest wheel optax jaxlib pytreeclass coverage
+          python -m pip install git+https://github.com/ASEM000/PyTreeClass@v0.2noDataclass
+          python -m pip install pytest wheel optax jaxlib  coverage
       - name: Pytest Check
         run: |
           python -m pip install .

diff --git a/.gitignore b/.gitignore
@@ -15,3 +15,5 @@ TODO.txt
 .trunk/trunk.yaml
 
 build
+tests.ipynb
+tests.ipynb
diff --git a/.trunk/.gitignore b/.trunk/.gitignore
@@ -1,2 +1,3 @@
 *out
+*logs
 external
diff --git a/.trunk/actions b/.trunk/actions
@@ -0,0 +1 @@
+/Users/asem/.cache/trunk/repos/bd27af2305e38c4721ddf501ce1b1a60/actions
diff --git a/.trunk/notifications b/.trunk/notifications
@@ -0,0 +1 @@
+/Users/asem/.cache/trunk/repos/bd27af2305e38c4721ddf501ce1b1a60/notifications
diff --git a/.trunk/trunk.yaml b/.trunk/trunk.yaml
@@ -1,6 +1,6 @@
 version: 0.1
 cli:
-  version: 0.15.1-beta
+  version: 1.5.1
 lint:
   enabled:
     - black@22.3.0

diff --git a/README.md b/README.md
@@ -1,5 +1,5 @@
 <div align = "center">
-<img  width=300px src="assets/kernexlogo.svg" align="center">
+<img  width=400 src="assets/kernexlogo.svg" align="center">
 
 
 <h3 align="center">Differentiable Stencil computations in JAX </h2>

diff --git a/assets/kernexlogo.svg b/assets/kernexlogo.svg
diff --git a/kernex/__init__.py b/kernex/__init__.py
@@ -1,20 +1,16 @@
-from ._src.base import kernelOperation
-from ._src.map import baseKernelMap, kernelMap, offsetKernelMap
-from ._src.scan import baseKernelScan, kernelScan, offsetKernelScan
+from ._src.map import kernel_map, offset_kernel_map
+from ._src.scan import kernel_scan, offset_kernel_scan
 from .interface.kernel_interface import kmap, kscan, smap, sscan
 
 __all__ = (
     "kmap",
     "kscan",
     "smap",
     "sscan",
-    "kernelOperation",
-    "baseKernelMap",
-    "kernelMap",
-    "offsetKernelMap",
-    "baseKernelScan",
-    "kernelScan",
-    "offsetKernelScan",
+    "kernel_map",
+    "offset_kernel_map",
+    "kernel_scan",
+    "offset_kernel_scan",
 )
 
-__version__ = "0.1.2"
+__version__ = "0.1.3"
diff --git a/kernex/_src/base.py b/kernex/_src/base.py
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		/Users/asem/.cache/trunk/repos/bd27af2305e38c4721ddf501ce1b1a60/actions