Fix for roc_auc_score import condition (#1101)

intel · Dec 16, 2022 · 6c27da3 · 6c27da3
1 parent 14849ee
commit 6c27da3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/daal4py.cpp b/src/daal4py.cpp
@@ -856,7 +856,7 @@ void c_train_test_split(data_or_file & orig, data_or_file & train, data_or_file
 
 double c_roc_auc_score(data_or_file & y_true, data_or_file & y_test)
 {
-#if __INTEL_DAAL__ == 2021 && INTEL_DAAL_VERSION >= 20210200
+#if __INTEL_DAAL__ >= 2021 && INTEL_DAAL_VERSION >= 20210200
  const size_t col_true = y_true.table->getNumberOfColumns();
  const size_t row_true = y_true.table->getNumberOfRows();
  const size_t col_test = y_test.table->getNumberOfColumns();

diff --git a/src/daal4py.h b/src/daal4py.h
@@ -52,7 +52,7 @@ using daal::services::LibraryVersionInfo;
 #include "data_management/data/internal/finiteness_checker.h"
 #include "data_management/data/internal/train_test_split.h"
 
-#if __INTEL_DAAL__ == 2021 && INTEL_DAAL_VERSION >= 20210200
+#if __INTEL_DAAL__ >= 2021 && INTEL_DAAL_VERSION >= 20210200
  #include "data_management/data/internal/roc_auc_score.h"
 #endif
 #if __INTEL_DAAL__ >= 2021 && INTEL_DAAL_VERSION >= 20210600