python3Packages.etils: adjust dependencies

pkgs/development/python-modules/etils/default.nix

@@ -17,10 +17,17 @@
   jupyter,
   mediapy,
   numpy,
+  packaging,
+  protobuf,
+  fsspec,
   importlib-resources,
   typing-extensions,
   zipp,
   absl-py,
+  simple-parsing,
+  einops,
+  gcsfs,
+  s3fs,
   tqdm,
   dm-tree,
   jax,
@@ -44,23 +51,33 @@ buildPythonPackage rec {
   optional-dependencies = rec {
     array-types = enp;
     eapp = [
-      absl-py # FIXME package simple-parsing
+      absl-py
+      simple-parsing
     ] ++ epy;
     ecolab =
       [
         jupyter
         numpy
         mediapy
+        packaging
+        protobuf
       ]
       ++ enp
-      ++ epy;
+      ++ epy
+      ++ etree;
     edc = epy;
-    enp = [ numpy ] ++ epy;
+    enp = [
+      numpy
+      einops
+    ] ++ epy;
     epath = [
+      fsspec
       importlib-resources
       typing-extensions
       zipp
     ] ++ epy;
+    epath-gcs = [ gcsfs ] ++ epath;
+    epath-s3 = [ s3fs ] ++ epath;
     epy = [ typing-extensions ];
     etqdm = [
       absl-py
@@ -70,6 +87,7 @@ buildPythonPackage rec {
     etree-dm = [ dm-tree ] ++ etree;
     etree-jax = [ jax ] ++ etree;
     etree-tf = [ tensorflow ] ++ etree;
+    lazy-imports = ecolab;
     all =
       array-types
       ++ eapp
@@ -77,6 +95,8 @@ buildPythonPackage rec {
       ++ edc
       ++ enp
       ++ epath
+      ++ epath-gcs
+      ++ epath-s3
       ++ epy
       ++ etqdm
       ++ etree