Fixing an oopsie

cmccomb · Jan 3, 2024 · 06c5f28 · 06c5f28
1 parent ce8cc56
commit 06c5f28
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/trussme/components.py b/trussme/components.py
@@ -518,7 +518,7 @@ def fos_yielding(self) -> float:
     @property
     def fos_buckling(self) -> float:
         """float: The factor of safety against buckling"""
-        return (
+        fos = (
             -(
                 (numpy.pi**2)
                 * self.elastic_modulus
@@ -527,3 +527,5 @@ def fos_buckling(self) -> float:
             )
             / self.force
         )
+
+        return fos if fos > 0 else numpy.inf
diff --git a/trussme/visualize.py b/trussme/visualize.py
@@ -51,7 +51,7 @@ def plot_truss(
 
     force_colormap = matplotlib.colors.LinearSegmentedColormap.from_list(
         "force",
-        [(1.0, 0.0, 0.0), (0.7, 0.7, 0.7), (0.0, 0.0, 1.0)],
+        numpy.array([[1.0, 0.0, 0.0], [0.7, 0.7, 0.7], [0.0, 0.0, 1.0]]),
     )
 
     for member in truss.members: