Upate

lanl-ansi · Oct 19, 2024 · 11b285d · 11b285d
1 parent f80db46
commit 11b285d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 10 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -36,9 +36,6 @@ jobs:
         run: |
            pip3 install torch --index-url https://download.pytorch.org/whl/cpu
            pip3 install numpy
-           python3 -c 'import torch'
-           which python
-           which python3
       # Install Julia
       - uses: julia-actions/setup-julia@v2
         with:

diff --git a/.github/workflows/documentation.yml b/.github/workflows/documentation.yml
@@ -15,7 +15,8 @@ jobs:
     env:
       GKSwstype: nul
       DATADEPS_ALWAYS_ACCEPT: true
-      JULIA_CONDAPKG_BACKEND: Null
+      JULIA_CONDAPKG_BACKEND: "Null"
+      JULIA_PYTHONCALL_EXE: "python3"
     steps:
       - uses: actions/checkout@v4
       # Install pytorch
@@ -25,7 +26,8 @@ jobs:
           python-version: '3.10'
       - name: Install pytorch
         run: |
-           pip3 install torch --index-url https://download.pytorch.org/whl/cpu
+          pip3 install torch --index-url https://download.pytorch.org/whl/cpu
+          pip3 install numpy
       # Install Julia
       - uses: julia-actions/setup-julia@latest
         with:

diff --git a/docs/make.jl b/docs/make.jl
@@ -37,9 +37,6 @@ function _literate_directory(dir)
         rm(filename)
     end
     for filename in _file_list(dir, dir, ".jl")
-        if endswith(filename, "pytorch.jl")
-            continue  # Skip for now
-        end
         # `include` the file to test it before `#src` lines are removed. It is
         # in a testset to isolate local variables between files.
         Test.@testset "$(filename)" begin
@@ -80,14 +77,14 @@ Documenter.makedocs(;
             "manual/Flux.md",
             "manual/GLM.md",
             "manual/Lux.md",
-            # "manual/PyTorch.md",
+            "manual/PyTorch.md",
         ],
         "Tutorials" => [
             "tutorials/student_enrollment.md",
             "tutorials/decision_trees.md",
             "tutorials/mnist.md",
             "tutorials/mnist_lux.md",
-            # "tutorials/pytorch.md",
+            "tutorials/pytorch.md",
             "tutorials/gaussian.md",
         ],
         "Developers" => ["developers/design_principles.md"],