Drop categories

Author: thodson-usgs

dataretrieval/waterdata/api.py

@@ -165,8 +165,7 @@ def get_daily(
         if your internet connection is spotty. The default (NA) will set the
         limit to the maximum allowable limit for the service.
     convert_type : boolean, optional
-        If True, converts columns to appropriate types like numeric, datetime,
-        boolean, categorical, etc.
+        If True, converts columns to appropriate types.
 
     Returns
     -------
@@ -476,8 +475,7 @@ def get_monitoring_locations(
         Note that the USGS Water Data APIs use camelCase "skipGeometry" in
         CQL2 queries.
     convert_type : boolean, optional
-        If True, converts columns to appropriate types like numeric, datetime,
-        boolean, categorical, etc.
+        If True, converts columns to appropriate types.
 
     Returns
     -------
@@ -669,8 +667,7 @@ def get_time_series_metadata(
         if your internet connection is spotty. The default (None) will set the
         limit to the maximum allowable limit for the service.
     convert_type : boolean, optional
-        If True, converts columns to appropriate types like numeric, datetime,
-        boolean, categorical, etc.
+        If True, converts columns to appropriate types.
 
     Returns
     -------
@@ -845,8 +842,7 @@ def get_latest_continuous(
         if your internet connection is spotty. The default (None) will set the
         limit to the maximum allowable limit for the service.
     convert_type : boolean, optional
-        If True, converts columns to appropriate types like numeric, datetime,
-        boolean, categorical, etc.
+        If True, converts columns to appropriate types.
 
     Returns
     -------
@@ -1020,8 +1016,7 @@ def get_latest_daily(
         if your internet connection is spotty. The default (None) will set the
         limit to the maximum allowable limit for the service.
     convert_type : boolean, optional
-        If True, converts columns to appropriate types like numeric, datetime,
-        boolean, categorical, etc.
+        If True, converts columns to appropriate types.
 
     Returns
     -------
@@ -1186,8 +1181,7 @@ def get_field_measurements(
         if your internet connection is spotty. The default (None) will set the
         limit to the maximum allowable limit for the service.
     convert_type : boolean, optional
-        If True, converts columns to appropriate types like numeric, datetime,
-        boolean, categorical, etc.
+        If True, converts columns to appropriate types.
 
     Returns
     -------

dataretrieval/waterdata/utils.py

@@ -703,22 +703,13 @@ def _type_cols(df: pd.DataFrame) -> pd.DataFrame:
         "last_modified",
         "time",
         ]
-    categorical_cols = [
-        "approval_status",
-        "monitoring_location_id",
-        "parameter_code",
-        "unit_of_measure",
-        ]
 
     for col in cols.intersection(time_cols):
         df[col] = pd.to_datetime(df[col], errors="coerce")
 
     for col in cols.intersection(numerical_cols):
         df[col] = pd.to_numeric(df[col], errors="coerce")
 
-    for col in cols.intersection(categorical_cols):
-        df[col] = df[col].astype("category")
-
     return df