Chore: Many small corrections and Nada Numpy update to 2.0.0 (#52)

* chore: updated numpy to version 2.0.0

* chore: added optional dependencies for examples and linting

* fix: added .gitignore to prevent not found file errors

* fix: Added linear_regression to testing infrastructure

* chore: bump version number

Author: jcabrero

examples/broadcasting/main.py

@@ -7,9 +7,9 @@
 
 import asyncio
 
-import nada_numpy.client as na_client
 import numpy as np
 import py_nillion_client as nillion
+from common.utils import compute, store_program, store_secret_array
 from config import DIM
 from cosmpy.aerial.client import LedgerClient
 from cosmpy.aerial.wallet import LocalWallet
@@ -20,7 +20,7 @@
                                     get_quote_and_pay, pay_with_quote)
 from py_nillion_client import NodeKey, UserKey
 
-from common.utils import compute, store_program, store_secret_array
+import nada_numpy.client as na_client
 
 home = os.getenv("HOME")
 load_dotenv(f"{home}/.config/nillion/nillion-devnet.env")

examples/broadcasting/src/broadcasting.py

@@ -2,11 +2,12 @@
 
 from typing import List
 
-# Step 0: Nada Numpy is imported with this line
-import nada_numpy as na
 from config import DIM
 from nada_dsl import Output, SecretInteger
 
+# Step 0: Nada Numpy is imported with this line
+import nada_numpy as na
+
 
 def nada_main() -> List[Output]:
     """

examples/common/utils.py

@@ -4,8 +4,6 @@
 import time
 from typing import Any, Callable, Dict, List
 
-import nada_numpy as na
-import nada_numpy.client as na_client
 import numpy as np
 import py_nillion_client as nillion
 from cosmpy.aerial.client import LedgerClient
@@ -14,6 +12,9 @@
                                     create_payments_config, get_quote,
                                     get_quote_and_pay, pay_with_quote)
 
+import nada_numpy as na
+import nada_numpy.client as na_client
+
 
 def async_timer(file_path: os.PathLike) -> Callable:
     """

examples/dot_product/main.py

@@ -7,9 +7,9 @@
 
 import asyncio
 
-import nada_numpy.client as na_client
 import numpy as np
 import py_nillion_client as nillion
+from common.utils import compute, store_program, store_secret_array
 from config import DIM
 from cosmpy.aerial.client import LedgerClient
 from cosmpy.aerial.wallet import LocalWallet
@@ -20,11 +20,10 @@
                                     get_quote_and_pay, pay_with_quote)
 from py_nillion_client import NodeKey, UserKey
 
-from common.utils import compute, store_program, store_secret_array
+import nada_numpy.client as na_client
 
 home = os.getenv("HOME")
 load_dotenv(f"{home}/.config/nillion/nillion-devnet.env")
-# load_dotenv(f"/workspaces/ai/.nillion-testnet.env")
 
 
 # Main asynchronous function to coordinate the process

examples/dot_product/src/dot_product.py

@@ -3,10 +3,10 @@
 from typing import List
 
 from config import DIM
+from nada_dsl import Output, SecretInteger
 
 # Step 0: Nada Numpy is imported with this line
 import nada_numpy as na
-from nada_dsl import Output, SecretInteger
 
 
 def nada_main() -> List[Output]:

examples/linear-regression/cleartext.ipynb renamed to examples/linear_regression/cleartext.ipynb

@@ -7,9 +7,9 @@
 
 import asyncio
 
-import nada_numpy.client as na_client
 import numpy as np
 import py_nillion_client as nillion
+from common.utils import compute, store_program, store_secret_array
 from cosmpy.aerial.client import LedgerClient
 from cosmpy.aerial.wallet import LocalWallet
 from cosmpy.crypto.keypairs import PrivateKey
@@ -20,12 +20,10 @@
 from py_nillion_client import NodeKey, UserKey
 from sklearn.linear_model import Ridge
 
-from common.utils import compute, store_program, store_secret_array
+import nada_numpy.client as na_client
 
 home = os.getenv("HOME")
-
 load_dotenv(f"{home}/.config/nillion/nillion-devnet.env")
-# load_dotenv(f"/workspaces/ai/.nillion-testnet.env")
 
 
 # Main asynchronous function to coordinate the process

examples/linear-regression/main.py renamed to examples/linear_regression/main.py

@@ -1,6 +1,7 @@
-import nada_numpy as na
 import numpy as np
 from nada_dsl import *
+
+import nada_numpy as na
 from nada_numpy.array import NadaArray
 
 

examples/linear-regression/nada-project.toml renamed to examples/linear_regression/nada-project.toml

@@ -1,8 +1,9 @@
-import nada_numpy as na
 import numpy as np
 from modular_inverse import PRIME, public_modular_inverse
 from nada_dsl import *
 
+import nada_numpy as na
+
 # from nada_crypto import random_lu_matrix, public_modular_inverse
 
 

examples/linear-regression/src/determinant.py renamed to examples/linear_regression/src/determinant.py

@@ -1,10 +1,11 @@
-import nada_numpy as na
 import numpy as np
 from determinant import determinant
 from gauss_jordan import gauss_jordan_zn
 from modular_inverse import (PRIME, private_modular_inverse,
                              public_modular_inverse)
 from nada_dsl import *
+
+import nada_numpy as na
 from nada_numpy.array import NadaArray
 
 

examples/linear-regression/src/gauss_jordan.py renamed to examples/linear_regression/src/gauss_jordan.py

@@ -1,6 +1,7 @@
-import nada_numpy as na
 import numpy as np
 from nada_dsl import *
+
+import nada_numpy as na
 from nada_numpy.array import NadaArray
 
 # LOG_SCALE = 8

examples/linear-regression/src/linear_regression.py renamed to examples/linear_regression/src/linear_regression.py

@@ -1,8 +1,9 @@
 import time
 
-import nada_numpy as na
 import numpy as np
 from nada_dsl import *
+
+import nada_numpy as na
 from nada_numpy.array import NadaArray
 
 LOG_SCALE = 16

examples/linear-regression/src/linear_regression_256.py renamed to examples/linear_regression/src/linear_regression_256.py

@@ -1,7 +1,8 @@
-import nada_numpy as na
 import numpy as np
 from nada_dsl import *
 
+import nada_numpy as na
+
 LOG_SCALE = 16
 SCALE = 1 << LOG_SCALE
 PRIME_64 = 18446744072637906947

examples/linear-regression/src/matrix_inverse.py renamed to examples/linear_regression/src/matrix_inverse.py

@@ -0,0 +1,5 @@
+# This directory is kept purposely, so that no compilation errors arise.
+# Ignore everything in this directory
+*
+# Except this file
+!.gitignore

examples/linear-regression/src/modular_inverse.py renamed to examples/linear_regression/src/modular_inverse.py

@@ -0,0 +1,38 @@
+---
+program: linear_regression
+inputs:
+  A_0_0:
+      SecretInteger: "-256"
+  A_0_1:
+      SecretInteger: "-199"
+  A_0_2:
+      SecretInteger: "-142"
+  A_1_0:
+      SecretInteger: "-85"
+  A_1_1:
+      SecretInteger: "-28"
+  A_1_2:
+      SecretInteger: "28"
+  A_2_0:
+      SecretInteger: "85"
+  A_2_1:
+      SecretInteger: "142"
+  A_2_2:
+      SecretInteger: "256"
+  b_0:
+      SecretInteger: "-256"
+  b_1:
+      SecretInteger: "-199"
+  b_2:
+      SecretInteger: "-142"
+#   lambda_0: 
+#     Integer: "1"
+expected_outputs:
+  b:
+    SecretInteger: "330643400256"
+  w_0:
+    SecretInteger: "1102041164640"
+  w_1:
+    SecretInteger: "-993683999568"
+  w_2:
+    SecretInteger: "1868226984"

examples/linear_regression/target/.gitignore

@@ -7,9 +7,9 @@
 
 import asyncio
 
-import nada_numpy.client as na_client
 import numpy as np
 import py_nillion_client as nillion
+from common.utils import compute, store_program, store_secret_array
 from config import DIM
 from cosmpy.aerial.client import LedgerClient
 from cosmpy.aerial.wallet import LocalWallet
@@ -20,11 +20,10 @@
                                     get_quote_and_pay, pay_with_quote)
 from py_nillion_client import NodeKey, UserKey
 
-from common.utils import compute, store_program, store_secret_array
+import nada_numpy.client as na_client
 
 home = os.getenv("HOME")
 load_dotenv(f"{home}/.config/nillion/nillion-devnet.env")
-# load_dotenv(f"/workspaces/ai/.nillion-testnet.env")
 
 
 # Main asynchronous function to coordinate the process

examples/linear-regression/tests/determinant-3.yaml renamed to examples/linear_regression/tests/determinant-3.yaml

@@ -2,11 +2,12 @@
 
 from typing import List
 
-# Step 0: Nada Numpy is imported with this line
-import nada_numpy as na
 from config import DIM
 from nada_dsl import Output, SecretInteger
 
+# Step 0: Nada Numpy is imported with this line
+import nada_numpy as na
+
 
 def nada_main() -> List[Output]:
     """

examples/linear-regression/tests/determinant_1.yaml renamed to examples/linear_regression/tests/determinant_1.yaml

@@ -7,9 +7,9 @@
 
 import asyncio
 
-import nada_numpy.client as na_client
 import numpy as np
 import py_nillion_client as nillion
+from common.utils import compute, store_program, store_secret_value
 from cosmpy.aerial.client import LedgerClient
 from cosmpy.aerial.wallet import LocalWallet
 from cosmpy.crypto.keypairs import PrivateKey
@@ -18,11 +18,10 @@
                                     create_payments_config)
 from py_nillion_client import NodeKey, UserKey
 
-from common.utils import compute, store_program, store_secret_value
+import nada_numpy.client as na_client
 
 home = os.getenv("HOME")
 load_dotenv(f"{home}/.config/nillion/nillion-devnet.env")
-# load_dotenv(f"/workspaces/ai/.nillion-testnet.env")
 
 
 # Main asynchronous function to coordinate the process

examples/linear-regression/tests/determinant_2.yaml renamed to examples/linear_regression/tests/determinant_2.yaml

@@ -2,9 +2,10 @@
 
 from typing import List
 
-import nada_numpy as na
 from nada_dsl import Output
 
+import nada_numpy as na
+
 
 def nada_main() -> List[Output]:
     """

examples/linear-regression/tests/gauss_jordan.yaml renamed to examples/linear_regression/tests/gauss_jordan.yaml

@@ -5,7 +5,7 @@
 
 # pylint:disable=too-many-lines
 
-from typing import Any, Callable, Optional, Sequence, Union, get_args, overload
+from typing import Any, Callable, Optional, Sequence, Union, get_args
 
 import numpy as np
 from nada_dsl import (Boolean, Input, Integer, Output, Party, PublicInteger,
@@ -870,28 +870,6 @@ def item(self, *args, **kwargs):
             return NadaArray(result)
         return result
 
-    @overload
-    def itemset(self, value: Any): ...
-    @overload
-    def itemset(self, item: Any, value: Any): ...
-
-    # pylint:disable=missing-function-docstring
-    @copy_metadata(np.ndarray.itemset)
-    def itemset(self, *args, **kwargs):
-        value = None
-        if len(args) == 1:
-            value = args[0]
-        elif len(args) == 2:
-            value = args[1]
-        else:
-            value = kwargs["value"]
-
-        _check_type_compatibility(value, self.dtype)
-        result = self.inner.itemset(*args, **kwargs)
-        if isinstance(result, np.ndarray):
-            return NadaArray(result)
-        return result
-
     # pylint:disable=missing-function-docstring
     @copy_metadata(np.ndarray.prod)
     def prod(self, *args, **kwargs):