Merge pull request #597 from Sichao25/cluster

Debug find_fixed_points
aristoteleo · Nov 5, 2023 · 7e2beee · 7e2beee
2 parents 9a687cc + 7066180
commit 7e2beee
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dynamo/vectorfield/scVectorField.py b/dynamo/vectorfield/scVectorField.py
@@ -599,18 +599,18 @@ def find_fixed_points(
         self,
         n_x0: int = 100,
         X0: Optional[np.ndarray] = None,
-        domain=None,
-        sampling_method=Literal["random", "velocity", "trn", "kmeans"],
+        domain: Optional[np.ndarray] = None,
+        sampling_method: Literal["random", "velocity", "trn", "kmeans"] = "random",
         **kwargs,
-    ):
+    ) -> None:
         """
         Search for fixed points of the vector field function.
 
         Args:
             n_x0: Number of sampling points
             X0: An array of shape (n_samples, n_dim)
             domain: Domain in which to search for fixed points
-            sampling_method:
+            sampling_method: Method for sampling initial points. Can be one of `random`, `velocity`, `trn`, or `kmeans`.
         """
         if domain is not None:
             domain = np.atleast_2d(domain)