fix docstring

FluxML · Nov 4, 2024 · 693a36f · 693a36f
1 parent 0a36651
commit 693a36f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/layers/recurrent.jl b/src/layers/recurrent.jl
@@ -237,6 +237,7 @@ julia> h′, c′ = l(x, (h, c));
 
 julia> size(h′)  # out x batch_size
 (5, 4)
+```
 """
 struct LSTMCell{I,H,V}
   Wi::I

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -11,7 +11,7 @@ using Functors: fmapstructure_with_path
 
 ## Uncomment below to change the default test settings
 # ENV["FLUX_TEST_AMDGPU"] = "true"
-# ENV["FLUX_TEST_CUDA"] = "true"
+ENV["FLUX_TEST_CUDA"] = "true"
 # ENV["FLUX_TEST_METAL"] = "true"
 # ENV["FLUX_TEST_CPU"] = "false"
 # ENV["FLUX_TEST_DISTRIBUTED_MPI"] = "true"
-Original file line number
+Diff line change
@@ Expand Up / @@ -237,6 +237,7 @@ julia> h′, c′ = l(x, (h, c)); @@
     julia> size(h′)  # out x batch_size
     (5, 4)
+    ```
     """
     struct LSTMCell{I,H,V}
       Wi::I
@@ Expand Down @@