Add docstring

javidahmed64592 · Apr 11, 2024 · b229291 · b229291
1 parent 6d7cbe7
commit b229291
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/math/matrix.py b/src/math/matrix.py
@@ -185,5 +185,9 @@ def map(matrix: Matrix, func: Callable) -> Matrix:
     def to_array(self) -> List[float]:
         """
         Return Matrix as a list of floats.
+
+        Returns:
+            matrix_list (List[float]): Matrix as list of floats
         """
-        return cast(List[float], self.data.tolist()[0])
+        matrix_list = self.data.tolist()[0]
+        return cast(List[float], matrix_list)