diff --git a/python/cuml/cuml/internals/array.py b/python/cuml/cuml/internals/array.py index 71918f77f5..8e07b5f89a 100644 --- a/python/cuml/cuml/internals/array.py +++ b/python/cuml/cuml/internals/array.py @@ -1,5 +1,5 @@ # -# Copyright (c) 2020-2024, NVIDIA CORPORATION. +# Copyright (c) 2020-2025, NVIDIA CORPORATION. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -1082,7 +1082,7 @@ def from_input( index = CudfIndex(index) if isinstance(X, CudfSeries): - if X.null_count != 0: + if X.null_count != 0 or bool(X.isnull().any()): raise ValueError( "Error: cuDF Series has missing/null values, " "which are not supported by cuML."