fixes a type hint

peterdsharpe · Jul 25, 2024 · cee8623 · cee8623
1 parent 0e802de
commit cee8623
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/aerosandbox/numpy/surrogate_model_tools.py b/aerosandbox/numpy/surrogate_model_tools.py
@@ -229,7 +229,10 @@ def sigmoid(
     return s_normalized
 
 
-def swish(x, beta=1):
+def swish(
+    x: Union[float, _np.ndarray],
+    beta: float = 1.0,
+):
     """
     A smooth approximation of the ReLU function, applied elementwise to an array `x`.