added elif

Author: 5hv5hvnk

pymc/distributions/shape_utils.py

@@ -451,28 +451,26 @@ def convert_shape(shape: Shape) -> Optional[WeakShape]:
         return None
     if isinstance(shape, int) or (isinstance(size, TensorVariable) and shape.ndim == 0):
         shape = (shape,)
-    if isinstance(shape, TensorVariable) and shape.ndim == 1:
+    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)}"
         )
-
     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}"
         )
-
     return shape
 
 
 def convert_size(size: Size) -> Optional[StrongSize]:
     """Process a user-provided size variable into None or a valid size object."""
     if size is None:
         return None
-    if isinstance(size, int) or (isinstance(size, TensorVariable) and size.ndim == 0):
+    elif isinstance(size, int) or (isinstance(size, TensorVariable) and size.ndim == 0):
         size = (size,)
     elif isinstance(size, TensorVariable) and size.ndim == 1:
         size = tuple(size)