From 9074d3bee74848016a8cee484bac236bee6e194f Mon Sep 17 00:00:00 2001
From: Sheng Zhong <zhsh@umich.edu>
Date: Mon, 8 Jul 2024 14:20:36 -0400
Subject: [PATCH] Add dependency

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index 551b7bc..1367724 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -48,6 +48,7 @@ dependencies = [
     'torch',
     'matplotlib',
     'pytorch_seed',
+    'arm_pytorch_utilities',
 ]
 
 [project.optional-dependencies]