docs: fix docs build failing

abess-team · Oct 1, 2023 · 43a4195 · 43a4195
1 parent 7af52d5
commit 43a4195
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.readthedocs.yaml b/.readthedocs.yaml
@@ -10,6 +10,8 @@ build:
       - pip install --no-cache-dir pybind11[global]
     pre_build:
       - pip install --no-cache-dir -r docs/requirements.txt
+  apt_packages:
+    - graphviz
 
 python:
   install:

diff --git a/docs/Tutorial/5-scikit-learn-connection/plot_2_geomstats.py b/docs/Tutorial/5-scikit-learn-connection/plot_2_geomstats.py
@@ -53,7 +53,7 @@
 data = np.concatenate((data0,data1))
 train_data, test_data, train_labels, test_labels = train_test_split(data, labels, test_size=0.33, random_state=0)
 
-mean = FrechetMean(metric=sphere.metric)
+mean = FrechetMean(sphere)
 mean.fit(train_data)
 mean_estimate = mean.estimate_