From ec759fdf705932df3f2131fd40a72102545f2bf2 Mon Sep 17 00:00:00 2001
From: Mit Kotak <53411468+mitkotak@users.noreply.github.com>
Date: Thu, 25 Jul 2024 16:55:53 -0700
Subject: [PATCH 1/2] Create README.md

---
 README.md | 8 ++++++++
 1 file changed, 8 insertions(+)
 create mode 100644 README.md

diff --git a/README.md b/README.md
new file mode 100644
index 0000000..42ea0f6
--- /dev/null
+++ b/README.md
@@ -0,0 +1,8 @@
+## TODO
+
+- [ ] Get `torch.export` pipeline working.
+- [ ] Train tetris
+- [ ] Port `allegro-jax`
+
+
+

From 7d5a1524fa9ab6ab8e305236e5d6c51f1cf477dc Mon Sep 17 00:00:00 2001
From: Taufeq Razakh <razakh@usc.edu>
Date: Fri, 26 Jul 2024 09:13:22 -0700
Subject: [PATCH 2/2] Minor fix to parameter name for scatter_mean when used in
 SimpleNetwork (#2)

* Minor fix to parameter name for scatter_mean when used in SimpleNetwork

* Add newline
---
 model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model.py b/model.py
index dc872f9..363869a 100644
--- a/model.py
+++ b/model.py
@@ -136,5 +136,5 @@ def forward(self,
         # Filter out 0e
         node_features = self.filter_tp(node_features, self.dummy_input)
 
-        graph_globals = scatter_mean(node_features, output_dim=[num_nodes])
-        return self.readout_mlp(graph_globals)
\ No newline at end of file
+        graph_globals = scatter_mean(node_features, dim=[num_nodes])
+        return self.readout_mlp(graph_globals)