Update shape_utils.py

Author: 5hv5hvnk

pymc/distributions/shape_utils.py

@@ -449,20 +449,16 @@ def convert_shape(shape: Shape) -> Optional[WeakShape]:
     """Process a user-provided shape variable into None or a valid shape object."""
     if shape is None:
         return None
-    if isinstance(shape, int) or (isinstance(size, TensorVariable) and shape.ndim == 0):
+    elif isinstance(shape, int) or (isinstance(size, TensorVariable) and shape.ndim == 0):
         shape = (shape,)
     elif isinstance(shape, TensorVariable) and shape.ndim == 1:
         shape = tuple(shape)
     elif isinstance(shape, (list, tuple)):
         shape = tuple(shape)
     else:
-        raise ValueError(
-            f"The `shape` parameter must be a tuple, TensorVariable, int or list. Actual: {type(shape)}"
-        )
+        raise ValueError(f"The `shape` parameter must be a tuple, TensorVariable, int or list. Actual: {type(shape)}")
     if isinstance(shape, tuple) and any(s == Ellipsis for s in shape[:-1]):
-        raise ValueError(
-            f"Ellipsis in `shape` may only appear in the last position. Actual: {shape}"
-        )
+        raise ValueError(f"Ellipsis in `shape` may only appear in the last position. Actual: {shape}")
     return shape
 
 
@@ -477,9 +473,7 @@ def convert_size(size: Size) -> Optional[StrongSize]:
     elif isinstance(size, (list, tuple)):
         size = tuple(size)
     else:
-        raise ValueError(
-            f"The `size` parameter must be a tuple, TensorVariable, int or list. Actual: {type(size)}"
-        )
+        raise ValueError(f"The `size` parameter must be a tuple, TensorVariable, int or list. Actual: {type(size)}")
     if isinstance(size, tuple) and Ellipsis in size:
         raise ValueError(f"The `size` parameter cannot contain an Ellipsis. Actual: {size}")
     return size