From 7a3a7a017cc74f44e86e3d5e4b00833b011e88b9 Mon Sep 17 00:00:00 2001
From: Jake Awe <jawe@nvidia.com>
Date: Thu, 18 Apr 2024 13:42:29 -0500
Subject: [PATCH] enable import tests

---
 ci/build_python.sh                | 8 +-------
 conda/recipes/cuproj/meta.yaml    | 4 +++-
 conda/recipes/cuspatial/meta.yaml | 2 ++
 3 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/ci/build_python.sh b/ci/build_python.sh
index a0ac89948..6d056e13d 100755
--- a/ci/build_python.sh
+++ b/ci/build_python.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright (c) 2022, NVIDIA CORPORATION.
+# Copyright (c) 2022-2024, NVIDIA CORPORATION.
 
 set -euo pipefail
 
@@ -27,19 +27,13 @@ CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
 
 rapids-logger "Begin py build cuSpatial"
 
-# TODO: Remove `--no-test` flag once importing on a CPU
-# node works correctly
 RAPIDS_PACKAGE_VERSION=${version} rapids-conda-retry mambabuild \
-  --no-test \
   --channel "${CPP_CHANNEL}" \
   conda/recipes/cuspatial
 
 rapids-logger "Begin py build cuProj"
 
-# TODO: Remove `--no-test` flag once importing on a CPU
-# node works correctly
 RAPIDS_PACKAGE_VERSION=${version} rapids-conda-retry mambabuild \
-  --no-test \
   --channel "${CPP_CHANNEL}" \
   conda/recipes/cuproj
 
diff --git a/conda/recipes/cuproj/meta.yaml b/conda/recipes/cuproj/meta.yaml
index 22309116e..1d6104010 100644
--- a/conda/recipes/cuproj/meta.yaml
+++ b/conda/recipes/cuproj/meta.yaml
@@ -1,4 +1,4 @@
-# Copyright (c) 2018-2023, NVIDIA CORPORATION.
+# Copyright (c) 2018-2024, NVIDIA CORPORATION.
 
 {% set version = environ['RAPIDS_PACKAGE_VERSION'].lstrip('v') %}
 {% set minor_version = version.split('.')[0] + '.' + version.split('.')[1] %}
@@ -84,6 +84,8 @@ test:            # [linux64]
     - cupy>=12.0.0
     - cuspatial ={{ minor_version }}
     - rmm ={{ minor_version }}
+  env:
+    - RAPIDS_NO_INITIALIZE=1
 
 
 about:
diff --git a/conda/recipes/cuspatial/meta.yaml b/conda/recipes/cuspatial/meta.yaml
index c2c60133d..320334af6 100644
--- a/conda/recipes/cuspatial/meta.yaml
+++ b/conda/recipes/cuspatial/meta.yaml
@@ -82,6 +82,8 @@ requirements:
 test:            # [linux64]
   imports:       # [linux64]
     - cuspatial  # [linux64]
+  env:
+    - RAPIDS_NO_INITIALIZE=1
 
 about:
   home: https://rapids.ai/