feat: pcgv3 ingest support

Author: dodamih

docker/Dockerfile.all_pcg

@@ -1,6 +1,6 @@
 # syntax=docker/dockerfile:1
 ARG PYTHON_VERSION=312
-ARG GRAPH_TOOL_VERSION="2.88"
+ARG GRAPH_TOOL_VERSION="2.97"
 ARG GRAPH_TOOL_MICROARCH="x86_64_v3"
 
 FROM "us-central1-docker.pkg.dev/zetta-research/zutils/zutils_base:cuda12.6_cudnn9.5.1_torch2.6.0_py${PYTHON_VERSION}"
@@ -18,7 +18,8 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
   --mount=type=cache,target=/root/.cache/pip,sharing=locked \
   savedAptMark="$(apt-mark showmanual)" \
   && cd /opt/zetta_utils \
-  && CONDA_OVERRIDE_ARCHSPEC=${GRAPH_TOOL_MICROARCH} conda install graph-tool-base=${GRAPH_TOOL_VERSION} \
+#  && CONDA_OVERRIDE_ARCHSPEC=${GRAPH_TOOL_MICROARCH} conda install -c conda-forge graph-tool-base=${GRAPH_TOOL_VERSION} \
+  && conda install -c conda-forge graph-tool-base=${GRAPH_TOOL_VERSION} \
   && ./install_zutils.py --dockerfile --mode modules --pcg \
   && rm -rf /root/.cache/bazel/ \
   && apt-mark auto '.*' > /dev/null \

install_zutils.py

@@ -611,7 +611,7 @@ def main():
     parser.add_argument(
         "--pcg", action="store_true", default=False, help="Whether or not to install PCG"
     )
-    parser.add_argument("--pcgtag", default="stitch_test_x6", help="PCG repo tag to use")
+    parser.add_argument("--pcgtag", default="v3_stitch_test_x3", help="PCG repo tag to use")
 
     args = parser.parse_args()
 

zetta_utils/common/__init__.py

@@ -10,4 +10,4 @@
 from .path import abspath, is_local
 from .pprint import lrpad
 from .signal_handlers import custom_signal_handler_ctx
-from .timer import RepeatTimer
+from .timer import RepeatTimer, Timer

zetta_utils/common/timer.py

@@ -1,4 +1,22 @@
+from __future__ import annotations
+
 import threading
+import time
+
+import attrs
+
+
+@attrs.mutable
+class Timer:  # pragma: no cover
+    start: float = 0.0
+    elapsed: float = 0.0
+
+    def __enter__(self):
+        self.start = time.perf_counter()
+        return self
+
+    def __exit__(self, *args):
+        self.elapsed = time.perf_counter() - self.start
 
 
 class RepeatTimer(threading.Timer):