use select_dtype

biocore · May 9, 2022 · 309c644 · 309c644
1 parent e76661a
commit 309c644
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/evident/q2/_methods.py b/evident/q2/_methods.py
@@ -14,8 +14,7 @@ def _check_provided_univariate_data(sample_metadata, data_column):
     """Check if provided univariate data is valid."""
     if data_column not in sample_metadata.columns:
         raise ValueError(f"{data_column} not found in sample metadata.")
-    if sample_metadata[data_column].dtype.kind not in ["i", "f"]:
-        # i = integer, f = float
+    if data_column not in sample_metadata.select_dtypes("number"):
         raise ValueError("Values in data_column must be numeric.")