@@ -129,7 +129,7 @@ def check_condition_df(
129129 )
130130
131131 for column_name in req_cols :
132- if not np . issubdtype (df [column_name ].dtype , np . number ):
132+ if pd . api . types . is_string_dtype (df [column_name ].dtype ):
133133 assert_no_leading_trailing_whitespace (
134134 df [column_name ].values , column_name
135135 )
@@ -173,14 +173,14 @@ def check_measurement_df(
173173 _check_df (df , MEASUREMENT_DF_REQUIRED_COLS , "measurement" )
174174
175175 for column_name in MEASUREMENT_DF_REQUIRED_COLS :
176- if not np . issubdtype (df [column_name ].dtype , np . number ):
176+ if pd . api . types . is_string_dtype (df [column_name ].dtype ):
177177 assert_no_leading_trailing_whitespace (
178178 df [column_name ].values , column_name
179179 )
180180
181181 for column_name in MEASUREMENT_DF_OPTIONAL_COLS :
182- if column_name in df and not np . issubdtype (
183- df [column_name ].dtype , np . number
182+ if column_name in df and pd . api . types . is_string_dtype (
183+ df [column_name ].dtype
184184 ):
185185 assert_no_leading_trailing_whitespace (
186186 df [column_name ].values , column_name
@@ -243,7 +243,7 @@ def check_parameter_df(
243243 check_ids (df .index .values , kind = "parameter" )
244244
245245 for column_name in PARAMETER_DF_REQUIRED_COLS [1 :]: # 0 is PARAMETER_ID
246- if not np . issubdtype (df [column_name ].dtype , np . number ):
246+ if pd . api . types . is_string_dtype (df [column_name ].dtype ):
247247 assert_no_leading_trailing_whitespace (
248248 df [column_name ].values , column_name
249249 )
@@ -304,14 +304,14 @@ def check_observable_df(observable_df: pd.DataFrame) -> None:
304304 check_ids (observable_df .index .values , kind = "observable" )
305305
306306 for column_name in OBSERVABLE_DF_REQUIRED_COLS [1 :]:
307- if not np . issubdtype (observable_df [column_name ].dtype , np . number ):
307+ if pd . api . types . is_string_dtype (observable_df [column_name ].dtype ):
308308 assert_no_leading_trailing_whitespace (
309309 observable_df [column_name ].values , column_name
310310 )
311311
312312 for column_name in OBSERVABLE_DF_OPTIONAL_COLS :
313- if column_name in observable_df and not np . issubdtype (
314- observable_df [column_name ].dtype , np . number
313+ if column_name in observable_df and pd . api . types . is_string_dtype (
314+ observable_df [column_name ].dtype
315315 ):
316316 assert_no_leading_trailing_whitespace (
317317 observable_df [column_name ].values , column_name
0 commit comments