From 43a419584ce279aaf3657eb6e6cfd2373dbbb5e6 Mon Sep 17 00:00:00 2001
From: bbayukari <17bbayukari@gmail.com>
Date: Sun, 1 Oct 2023 14:34:25 +0800
Subject: [PATCH] docs: fix docs build failing

---
 .readthedocs.yaml                                           | 2 ++
 docs/Tutorial/5-scikit-learn-connection/plot_2_geomstats.py | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.readthedocs.yaml b/.readthedocs.yaml
index de5f0b28..bd11fd93 100644
--- 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
index 4b24da06..41f55d57 100644
--- 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_