chore: more fixes

NillionNetwork · Nov 27, 2024 · 87a1965 · 87a1965
1 parent f394956
commit 87a1965
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 7 deletions.
diff --git a/nada_dsl/nada_types/collections.py b/nada_dsl/nada_types/collections.py
@@ -108,7 +108,7 @@ def store_in_ast(self, ty):
         )
 
 
-class TupleType(DslType):
+class TupleType(NadaType):
     """Marker type for Tuples."""
 
     is_compound = True
@@ -171,7 +171,7 @@ def _generate_accessor(ty: Any, accessor: Any) -> DslType:
     return ty.instantiate(accessor)
 
 
-class NTupleType(DslType):
+class NTupleType(NadaType):
     """Marker type for NTuples."""
 
     is_compound = True
@@ -261,7 +261,7 @@ def store_in_ast(self, ty: object):
         )
 
 
-class ObjectType(DslType):
+class ObjectType(NadaType):
     """Marker type for Objects."""
 
     is_compound = True
@@ -412,7 +412,7 @@ def store_in_ast(self, ty: DslTypeRepr):
         )
 
 
-class ArrayType(DslType):
+class ArrayType(NadaType):
     """Marker type for arrays."""
 
     is_compound = True
@@ -533,7 +533,6 @@ def inner_product(self: "Array[T]", other: "Array[T]") -> T:
         if is_primitive_integer(self.contained_type) and is_primitive_integer(
             other.contained_type
         ):
-
             return self.contained_type.instantiate(
                 InnerProduct(left=self, right=other, source_ref=SourceRef.back_frame())
             )  # type: ignore

diff --git a/test-programs/ntuple_accessor.py b/test-programs/ntuple_accessor.py
@@ -26,5 +26,6 @@ def add(acc: PublicInteger, a: PublicInteger) -> PublicInteger:
 
     return [Output(final, "my_output", party1)]
 
+
 if __name__ == "__main__":
-    nada_main()
+    nada_main()
diff --git a/test-programs/object_accessor.py b/test-programs/object_accessor.py
@@ -15,7 +15,6 @@ def nada_main():
     array = object.b
     scalar_2 = object.c
 
-
     def add(acc: PublicInteger, a: PublicInteger) -> PublicInteger:
         return acc + a