fix(pcg): graph_tool import; remove ununed line in dockerfiles

ZettaAI · Jan 28, 2025 · 6550b7a · 6550b7a
1 parent 15a17a9
commit 6550b7a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/docker/Dockerfile.all b/docker/Dockerfile.all
@@ -7,7 +7,6 @@ ENV DEBIAN_FRONTEND="noninteractive"
 RUN rm -f /etc/apt/apt.conf.d/docker-clean; echo 'Binary::apt::APT::Keep-Downloaded-Packages "true";' > /etc/apt/apt.conf.d/keep-cache
 COPY pyproject.toml /opt/zetta_utils/
 COPY install_zutils.py /opt/zetta_utils/
-COPY compile_waterz /opt/zetta_utils/
 ENV PYTHONPATH=/opt/zetta_utils
 WORKDIR /opt/zetta_utils
 

diff --git a/docker/Dockerfile.all_pcg b/docker/Dockerfile.all_pcg
@@ -7,7 +7,6 @@ ENV DEBIAN_FRONTEND="noninteractive"
 RUN rm -f /etc/apt/apt.conf.d/docker-clean; echo 'Binary::apt::APT::Keep-Downloaded-Packages "true";' > /etc/apt/apt.conf.d/keep-cache
 COPY pyproject.toml /opt/zetta_utils/
 COPY install_zutils.py /opt/zetta_utils/
-COPY compile_waterz /opt/zetta_utils/
 ENV PYTHONPATH=/opt/zetta_utils
 WORKDIR /opt/zetta_utils
 

diff --git a/zetta_utils/__init__.py b/zetta_utils/__init__.py
@@ -1,8 +1,13 @@
-# pylint: disable=unused-import, import-outside-toplevel
+# pylint: disable=unused-import, import-outside-toplevel, broad-exception-caught, import-error
 """Zetta AI Computational Connectomics Toolkit."""
 import os
 import sys
 
+try:
+    import graph_tool
+except Exception:
+    ...
+
 from . import log, typing, parsing, builder, common, constants
 from . import geometry, distributions, layer, ng
 from .log import get_logger