use set with in for conditional statement

bambinos · Sep 15, 2023 · abf9c0d · abf9c0d
1 parent c045861
commit abf9c0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bambi/interpret/utils.py b/bambi/interpret/utils.py
@@ -341,7 +341,7 @@ def set_default_values(model: Model, data_dict: dict, kind: str) -> dict:
             dtype = str(model.data[name].dtype)
             if re.match(r"float*|int*", dtype):
                 data_dict[name] = np.mean(model.data[name])
-            elif dtype == "category" or dtype == "object":
+            elif dtype in ("category", "dtype"):
                 data_dict[name] = mode(model.data[name])
 
     if kind in ("comparisons", "slopes"):