Added runtime error in case radii of torus do not match

aidos-lab · Dec 12, 2023 · 4bab952 · 4bab952
1 parent 82248dd
commit 4bab952
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/torch_topological/data/shapes.py b/torch_topological/data/shapes.py
@@ -162,6 +162,12 @@ def sample_from_torus(n, d=3, r=1.0, R=2.0, seed=None):
     torch.tensor of shape `(n, d)`
         Tensor of sampled coordinates.
     """
+    if r > R:
+        raise RuntimeError(
+            'Radius of the tube must be less than or equal to radius of the torus'
+        )
+
+
     rng = np.random.default_rng(seed)
     angles = []