From ad4e5a3d6f5f73374d1bb8ca64cdcf2e8d5f2181 Mon Sep 17 00:00:00 2001
From: bcicc <tyler@saguarocm.com>
Date: Sun, 21 Apr 2024 12:15:18 -0700
Subject: [PATCH] bump python, cuml, cuda bounds

---
 CondaPkg.toml    | 16 ++++++++--------
 src/CuDF/CuDF.jl |  4 ++--
 src/utils.jl     |  2 +-
 test/cuml.jl     | 14 +++++++-------
 4 files changed, 18 insertions(+), 18 deletions(-)

diff --git a/CondaPkg.toml b/CondaPkg.toml
index bcc4cfe..f62a68a 100644
--- a/CondaPkg.toml
+++ b/CondaPkg.toml
@@ -1,14 +1,14 @@
 channels = ["rapidsai", "nvidia", "conda-forge"]
 
 [deps]
-cudf = "=23.10"
-cuxfilter = "=23.10"
-cucim = "=23.10"
-cuspatial = "=23.10"
-cugraph = "=23.10"
-cuml = "=23.10"
-python = ">=3.9,<3.11"
+cudf = "=24.04"
+cuxfilter = "=24.04"
+cucim = "=24.04"
+cuspatial = "=24.04"
+cugraph = "=24.04"
+cuml = "=24.04"
+python = ">=3.9,<=3.11"
 
     [deps.cuda-version]
     channel = "conda-forge"
-    version = "=12.0"
+    version = "=12.2"
diff --git a/src/CuDF/CuDF.jl b/src/CuDF/CuDF.jl
index e0ec616..642076d 100644
--- a/src/CuDF/CuDF.jl
+++ b/src/CuDF/CuDF.jl
@@ -1,10 +1,10 @@
 
 module CuDF
 
-using Dates
+using Dates: Date
 using PythonCall
 using Lazy
-using Compat
+using Compat: Compat
 using TableTraits
 using Statistics
 
diff --git a/src/utils.jl b/src/utils.jl
index 1f94a37..0e0db6d 100644
--- a/src/utils.jl
+++ b/src/utils.jl
@@ -1,6 +1,6 @@
 
 # List of CUDA versions supported by RAPIDSAI
-const supported_versions = ["11.2", "11.8", "12.0"]
+const supported_versions = ["11.2", "11.8", "12.0", "12.2"]
 
 function find_closest_supported_version(major, supported_versions)
     major_versions = filter(v -> startswith(v, major), supported_versions)
diff --git a/test/cuml.jl b/test/cuml.jl
index 56e4e33..3bed40d 100644
--- a/test/cuml.jl
+++ b/test/cuml.jl
@@ -170,7 +170,7 @@ end
         model = PCA()
         mach = machine(model, X)
         fit!(mach)
-        X_trans = transform(mach, X)
+        X_trans = MLJBase.transform(mach, X)
         inverse_transform(mach, X)
     end
 
@@ -178,14 +178,14 @@ end
         model = IncrementalPCA()
         mach = machine(model, X)
         fit!(mach)
-        X_trans = transform(mach, X)
+        X_trans = MLJBase.transform(mach, X)
     end
 
     @testset "TruncatedSVD" begin
         model = TruncatedSVD(; n_components=2)
         mach = machine(model, X)
         fit!(mach)
-        X_trans = transform(mach, X)
+        X_trans = MLJBase.transform(mach, X)
         inverse_transform(mach, X)
     end
 
@@ -193,28 +193,28 @@ end
         model = UMAP(; n_components=2)
         mach = machine(model, X)
         fit!(mach)
-        X_trans = transform(mach, X)
+        X_trans = MLJBase.transform(mach, X)
     end
 
     @testset "GaussianRandomProjection" begin
         model = GaussianRandomProjection(; n_components=2)
         mach = machine(model, X)
         fit!(mach)
-        X_trans = transform(mach, X)
+        X_trans = MLJBase.transform(mach, X)
     end
 
     @testset "SparseRandomProjection" begin
         model = SparseRandomProjection(; n_components=2)
         mach = machine(model, X)
         fit!(mach)
-        X_trans = transform(mach, X)
+        X_trans = MLJBase.transform(mach, X)
     end
 
     @testset "TSNE" begin
         model = TSNE(; n_components=2)
         mach = machine(model, X)
         fit!(mach)
-        X_trans = transform(mach, X)
+        X_trans = MLJBase.transform(mach, X)
     end
 end