update to new pio API and use unreleased Plotly and Kaleido

emilykl · emilykl · commit 8c84e4bea2dd · 2025-05-29T15:33:30.000-04:00
diff --git a/.circleci/env_image.sh b/.circleci/env_image.sh
@@ -6,6 +6,10 @@ sudo python3 .circleci/download_google_fonts.py
 sudo cp -r .circleci/fonts/ /usr/share/
 sudo fc-cache -f
 # install kaleido & plotly
-sudo python3 -m pip install "plotly[kaleido]==6.1.2" --progress-bar off
+# sudo python3 -m pip install "plotly[kaleido]==6.1.2" --progress-bar off
+# Once new Kaleido and Plotly versions are released, uncomment the line above, update the Plotly version,
+# and delete the two lines below.
+sudo python3 -m pip install "git+https://github.com/plotly/plotly.py.git@6837831" --progress-bar off
+sudo python3 -m pip install "'git+https://github.com/plotly/Kaleido.git@2a4bfa2#subdirectory=src/py'" --progress-bar off
 # install numpy i.e. to convert arrays to typed arrays
 sudo python3 -m pip install numpy==1.24.2
diff --git a/test/image/make_baseline.py b/test/image/make_baseline.py
@@ -19,7 +19,7 @@
 dirIn = os.path.join(root, 'test', 'image', 'mocks')
 dirOut = os.path.join(root, 'build', 'test_images')
 
-# N.B. equal is the falg to write to baselines not test_images
+# N.B. equal is the flag to write to baselines not test_images
 
 if '=' in args :
     args = args[args.index('=') + 1:]
@@ -35,7 +35,7 @@
     # until https://github.com/plotly/Kaleido/issues/124 is addressed
     # we are uanble to use local mathjax v3 installed in node_modules
     # for now let's download it from the internet:
-    pio.kaleido.scope.mathjax = 'https://cdn.jsdelivr.net/npm/mathjax@3.2.2/es5/tex-svg.js'
+    pio.defaults.mathjax = 'https://cdn.jsdelivr.net/npm/mathjax@3.2.2/es5/tex-svg.js'
     mathjax_version = 3
     print('Kaleido using MathJax v3')
 
@@ -52,7 +52,8 @@
 
     plotlyjs = plotlyjs_with_virtual_webgl
 
-pio.kaleido.scope.plotlyjs = plotlyjs
+pio.defaults.plotlyjs = plotlyjs
+
 pio.templates.default = 'none'
 
 ALL_MOCKS = [os.path.splitext(a)[0] for a in os.listdir(dirIn) if a.endswith('.json')]
