bump python, cuml, cuda bounds

tylerjthomas9 · Apr 21, 2024 · ad4e5a3 · ad4e5a3
1 parent db03b13
commit ad4e5a3
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 18 deletions.
diff --git 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
@@ -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
@@ -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
@@ -170,51 +170,51 @@ 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
 
     @testset "IncrementalPCA" begin
         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
 
     @testset "UMAP" begin
         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