Updated docstring for Pt Linspace.

2025-03-01 13:00:05 -08:00
parent acd3032f84
commit 047ce6f8ad
3 changed files with 3 additions and 3 deletions
@@ -12,7 +12,7 @@ from cda.utils import str_to_number, str_to_number_with_default

 class PtLinspace:
    """
-    Creates a PyTorch tensor using `torch.arange` with the specified start, end, and steps values.
+    Creates a PyTorch tensor using `torch.linspace` with the specified start, end, and steps values.
    
    The start and end values are parsed from string inputs and converted to a float.
    The steps value is converted to an int.