.

Dockerfile

@@ -66,7 +66,7 @@ RUN python3 -m pip install fastjet
 RUN python3 -m pip install gradio
 RUN python3 -m pip install huggingface_hub
 RUN python3 -m pip install hdbscan
-RUN python3 -m pip install lgatr
+#RUN python3 -m pip install lgatr # This doesn't work
 RUN python3 -c "\
 from huggingface_hub import snapshot_download; \
 snapshot_download(repo_id='gregorkrzmanc/jetclustering', local_dir='models/'); \
@@ -84,9 +84,10 @@ RUN mkdir -p /opt/pepr
 
 
 # Install L-GATr - for some reason this only works if executed from the already-built container
-#RUN cd /opt/pepr && git clone https://github.com/gregorkrz/lorentz-gatr lgatr
-#RUN cd /opt/pepr/lgatr/ && python3 -m pip install .
-
+RUN cd /opt/pepr && git clone https://github.com/gregorkrz/lorentz-gatr lgatr
+RUN cd /opt/pepr/lgatr/ && python3 -m pip install .
+RUN ls /usr/local/lib/python3.10/dist-packages/lgatr
+RUN ls /usr/local/lib/python3.10/dist-packages/lgatr/layers
 # Install torch_cmspepr
 
 RUN cd /opt/pepr && git clone https://github.com/cms-pepr/pytorch_cmspepr