Skip to content

Commit 70e0c0c

Browse files
Fix RLZoo compat
1 parent 0cda415 commit 70e0c0c

File tree

2 files changed

+16
-3
lines changed

2 files changed

+16
-3
lines changed

src/ReinforcementLearningZoo/Project.toml

+4-2
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ ReinforcementLearningBase = "e575027e-6cd6-5018-9292-cdc6200d2b44"
1818
ReinforcementLearningCore = "de1b191a-4ae0-4afa-a27b-92d07f46b2d6"
1919
StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
2020
Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
21-
cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
2221

2322
[compat]
2423
CUDA = "4, 5"
@@ -38,7 +37,10 @@ cuDNN = "1"
3837
julia = "1.9"
3938

4039
[extras]
40+
cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
41+
Preferences = "21216c6a-2e73-6563-6e65-726566657250"
4142
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
43+
UUIDs = "cf7118a7-6976-5b1a-9a39-7adc72f591a4"
4244

4345
[targets]
44-
test = ["Test"]
46+
test = ["cuDNN", "Preferences", "Test", "UUIDs"]

src/ReinforcementLearningZoo/test/runtests.jl

+12-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,15 @@
1-
using Test
1+
using UUIDs
2+
using Preferences
3+
4+
if Sys.isapple()
5+
flux_uuid = UUID("587475ba-b771-5e3f-ad9e-33799f191a9c")
6+
set_preferences!(flux_uuid, "gpu_backend" => "Metal")
7+
8+
using Metal
9+
else
10+
using CUDA, cuDNN
11+
CUDA.allowscalar(false)
12+
end
213

314
@testset "ReinforcementLearningZoo.jl" begin
415
# include("cfr/cfr.jl")

0 commit comments

Comments
 (0)