[CI] Fix docs

ghstack-source-id: 803dfa1
Pull-Request-resolved: #1298

Author: Vincent Moens

.github/workflows/docs.yml

@@ -37,7 +37,7 @@ jobs:
       script: |
         set -e
         set -v
-        apt-get update && apt-get install -y -f git wget gcc g++ dialog apt-utils cmake
+        apt-get update && apt-get install -y -f git wget gcc g++ dialog apt-utils
         root_dir="$(pwd)"
         conda_dir="${root_dir}/conda"
         env_dir="${root_dir}/env"
@@ -55,6 +55,7 @@ jobs:
 
         # 2. upgrade pip, ninja and packaging
         apt-get install python3-pip unzip -y -f
+        conda install anaconda::cmake -y
         python3 -m pip install --upgrade pip
         python3 -m pip install setuptools ninja packaging "pybind11[global]" -U
 

tutorials/sphinx_tuto/export.py

@@ -107,7 +107,7 @@
 
 from torch.export import export
 
-model_export = export(model, args=(), kwargs={"x": x})
+model_export = export(model, args=(), kwargs={"x": x}, strict=True)
 
 ##################################################
 # Let us look at the module: