From 23a4e1cc7f3fad7982560568b52d6a03ddaa68d4 Mon Sep 17 00:00:00 2001
From: Kaustubh Tangsali <ktangsali@nvidia.com>
Date: Wed, 27 Nov 2024 21:51:19 +0000
Subject: [PATCH 1/2] remove numpy pin, update base pytorch container

---
 Dockerfile     | 6 +++---
 pyproject.toml | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 242cd71e3..b7067a9e5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -14,7 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-ARG BASE_CONTAINER=nvcr.io/nvidia/pytorch:24.06-py3
+ARG BASE_CONTAINER=nvcr.io/nvidia/pytorch:24.11-py3
 FROM ${BASE_CONTAINER} as builder
 
 ARG TARGETPLATFORM
@@ -88,10 +88,10 @@ RUN if [ "$TARGETPLATFORM" = "linux/arm64" ] && [ "$DGL_ARM64_WHEEL" != "unknown
         pip install --no-cache-dir --no-deps /modulus/deps/${DGL_ARM64_WHEEL}; \
     elif [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
         echo "Installing DGL for: $TARGETPLATFORM" && \
-        pip install --no-cache-dir --no-deps dgl==2.0.0 -f https://data.dgl.ai/wheels/cu121/repo.html; \
+        pip install --no-cache-dir --no-deps dgl -f https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html; \
     else \
         echo "No custom wheel or wheel on PyPi found. Installing DGL for: $TARGETPLATFORM from source" && \
-        git clone https://github.com/dmlc/dgl.git && cd dgl/ && git checkout tags/v2.0.0 && git submodule update --init --recursive && \
+        git clone https://github.com/dmlc/dgl.git && cd dgl/ && git checkout tags/v2.4.0 && git submodule update --init --recursive && \
         DGL_HOME="/workspace/dgl" bash script/build_dgl.sh -g && \
         cd python && \
         python setup.py install && \
diff --git a/pyproject.toml b/pyproject.toml
index 52807527d..24d4da7cf 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,7 +13,7 @@ requires-python = ">=3.10"
 license = {text = "Apache 2.0"}
 dependencies = [
     "torch>=2.0.0",
-    "numpy>=1.22.4,<1.25",
+    "numpy>=1.22.4",
     "xarray>=2023.1.0",
     "zarr>=2.14.2",
     "fsspec>=2023.1.0",

From dbff1af42962bbbc2f01ecb14aa292802968deb4 Mon Sep 17 00:00:00 2001
From: Kaustubh Tangsali <ktangsali@nvidia.com>
Date: Wed, 27 Nov 2024 21:53:04 +0000
Subject: [PATCH 2/2] update changelog

---
 CHANGELOG.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 295eebc3d..b5d5d74e0 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -22,6 +22,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Dependencies
 
+- Remove the numpy dependency upper bound.
+
 ## [0.9.0] - 2024-12-04
 
 ### Added