We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 197c477 commit d7bdfcfCopy full SHA for d7bdfcf
brml/prob_tables/const.py
@@ -44,7 +44,7 @@ def _check_table(self, value):
44
# If list or numpy array, check size
45
elif isinstance(value, (np.ndarray, list)):
46
# If only 1 element -> return flatten array
47
- if np.array([value]).size == 1:
+ if np.array([value]).size <= 1:
48
return np.array([value]).flatten()
49
# Otherwise provided more than one value
50
else:
0 commit comments