Merge pull request #376 from gdsfactory/update_flax

update jaxlib and flax
gdsfactory · Apr 10, 2024 · 89530a7 · 89530a7
2 parents 52fb11d + b71c876
commit 89530a7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/gplugins/path_length_analysis/path_length_analysis.py b/gplugins/path_length_analysis/path_length_analysis.py
@@ -423,8 +423,8 @@ def get_pathlength_widgets(
         k: node_positions[d["name"]] for k, d in graph_to_viz.nodes(data=True)
     }
     graph_renderer = from_networkx(graph_to_viz, node_positions_by_int_label)
-    graph_renderer.node_renderer.glyph = Circle(size=5, fill_color=Spectral4[0])
-    graph_renderer.node_renderer.hover_glyph = Circle(size=15, fill_color=Spectral4[1])
+    graph_renderer.node_renderer.glyph = Circle(fill_color=Spectral4[0])
+    graph_renderer.node_renderer.hover_glyph = Circle(fill_color=Spectral4[1])
 
     graph_renderer.edge_renderer.glyph = MultiLine(
         line_color="edge_color", line_alpha=0.8, line_width=5

diff --git a/pyproject.toml b/pyproject.toml
@@ -81,14 +81,16 @@ klayout = [
   "pyvis<=0.3.1"
 ]
 meow = [
-  "jax==0.4.23",
-  "jaxlib==0.4.23",
+  "jax>=0.4.26",
+  "jaxlib>=0.4.26",
+  "flax>=0.8.2",
   "meow-sim>=0.9.0,<0.11.0",
   "tidy3d==2.5.2"
 ]
 sax = [
-  "jax<=0.4.23",
-  "jaxlib<=0.4.23",
+  "jax>=0.4.26",
+  "jaxlib>=0.4.26",
+  "flax>=0.8.2",
   "sax>=0.12.1,<0.13.0",
   "scikit-learn",
   "pyvis<=0.3.1"