Use new jaxtype decorator convention

eigenvivek · Jan 10, 2024 · 7283f6a · 7283f6a
1 parent 30c1dd7
commit 7283f6a
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 8 deletions.
diff --git a/diffpose/deepfluoro.py b/diffpose/deepfluoro.py
@@ -306,8 +306,6 @@ def preprocess(img, size=None, initial_energy=torch.tensor(65487.0)):
     return img
 
 # %% ../notebooks/api/00_deepfluoro.ipynb 26
-from beartype import beartype
-
 from .calibration import RigidTransform, convert
 
 

diff --git a/diffpose/ljubljana.py b/diffpose/ljubljana.py
@@ -115,8 +115,6 @@ def __getitem__(self, idx):
         )
 
 # %% ../notebooks/api/01_ljubljana.ipynb 7
-from beartype import beartype
-
 from .calibration import RigidTransform, convert
 
 

diff --git a/notebooks/api/00_deepfluoro.ipynb b/notebooks/api/00_deepfluoro.ipynb
@@ -712,8 +712,6 @@
    "outputs": [],
    "source": [
     "#| export\n",
-    "from beartype import beartype\n",
-    "\n",
     "from diffpose.calibration import RigidTransform, convert\n",
     "\n",
     "\n",

diff --git a/notebooks/api/01_ljubljana.ipynb b/notebooks/api/01_ljubljana.ipynb
@@ -186,8 +186,6 @@
    "outputs": [],
    "source": [
     "#| export\n",
-    "from beartype import beartype\n",
-    "\n",
     "from diffpose.calibration import RigidTransform, convert\n",
     "\n",
     "\n",
-Original file line number
+Diff line change
@@ Expand Up / @@ -115,8 +115,6 @@ def __getitem__(self, idx): @@
             )
     # %% ../notebooks/api/01_ljubljana.ipynb 7
-    from beartype import beartype
     from .calibration import RigidTransform, convert
@@ Expand Down @@