Exemplo n.º 1
0
def hook():
    hook = TorchHook(torch)
    return hook
Exemplo n.º 2
0
from syft.frameworks.torch.tensors.interpreters.paillier import PaillierTensor
from syft.serde.serde import deserialize
from phe.paillier import EncryptedNumber
from phe.paillier import PaillierPublicKey
from json import JSONEncoder
from numpy import ndarray
from numpy import array
from numpy import random
from web3 import Web3

from syft import TorchHook
import torch
hook = TorchHook(torch)


# User manipulation functions
def retrieve_user_info(user_path='info/user_info.txt'):
    user_file = open(user_path, 'r')
    address = user_file.readline().lstrip('address:').rstrip()
    private_key = user_file.readline().lstrip('private_key:').rstrip()
    endpoint_url = user_file.readline().lstrip('endpoint_url:').rstrip()
    user_file.close()
    return address, private_key, endpoint_url


# Paillier tensor manipulation functions
class MultiDimensionalArrayEncoder(JSONEncoder):
    def encode(self, obj):
        def hint_tuples(item):
            if isinstance(item, tuple):
                return {'__tuple__': True, 'items': item}