diff --git a/mattport/nerf/graph.py b/mattport/nerf/graph.py index 389ed6867..f309a46fa 100644 --- a/mattport/nerf/graph.py +++ b/mattport/nerf/graph.py @@ -24,6 +24,7 @@ class Node: def __hash__(self): return hash(self.name) + class Graph(nn.Module): """_summary_""" diff --git a/tests/nerf/test_graph.py b/tests/nerf/test_graph.py index 00530a164..96a10d1ce 100644 --- a/tests/nerf/test_graph.py +++ b/tests/nerf/test_graph.py @@ -147,4 +147,4 @@ def test_construct_graph_complex(): assert len(roots) == 1 roots = list(roots) targets = [("encoder_0", 3), ("mlp_0", 2), ("mlp_3", 0)] - check_consistency(roots[0], targets) \ No newline at end of file + check_consistency(roots[0], targets)