diff --git a/pkgs/development/python-modules/etils/default.nix b/pkgs/development/python-modules/etils/default.nix index c192f670940f..0718b0abb60d 100644 --- a/pkgs/development/python-modules/etils/default.nix +++ b/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