import os import sys from logging import getLogger from tatau_core import settings from tatau_core.node.worker import Worker from tatau_core.utils.logging import configure_logging configure_logging('worker') logger = getLogger('tatau_core') def load_credentials(account_address_var_name): address = os.getenv(account_address_var_name) if address is None: raise ValueError('{} is not specified'.format(account_address_var_name)) storage_path = settings.KEYS_PATH dir_name = address.replace('0x', '') with open(os.path.join(storage_path, dir_name, 'rsa_pk.pem'), 'r') as f: pk = f.read() return address, pk.encode() if __name__ == '__main__': try: index = '_{}'.format(sys.argv[1]) settings.TATAU_STORAGE_BASE_DIR += str(index) except IndexError:
import os import sys from logging import getLogger from tatau_core import settings from tatau_core.contract import NodeContractInfo from tatau_core.node import VerifierEstimator from tatau_core.utils.logging import configure_logging configure_logging('verifier') logger = getLogger('tatau_core') def load_credentials(account_address_var_name): address = os.getenv(account_address_var_name) if address is None: raise ValueError( '{} is not specified'.format(account_address_var_name)) storage_path = settings.KEYS_PATH dir_name = address.replace('0x', '') with open(os.path.join(storage_path, dir_name, 'rsa_pk.pem'), 'r') as f: pk = f.read() with open(os.path.join(storage_path, dir_name, 'wallet.json'), 'r') as f: wallet = f.read() with open(os.path.join(storage_path, dir_name, 'wallet.pass'), 'r') as f: wallet_password = f.read()
import argparse from logging import getLogger from tatau_core.nn.benchmark.run import benchmark_train from tatau_core.utils.logging import configure_logging configure_logging(__name__) logger = getLogger('tatau_core') if __name__ == '__main__': parser = argparse.ArgumentParser(description='Local benchmark') parser.add_argument('-m', '--model_path', required=True, help='model path') parser.add_argument('-train', '--dataset_train', required=True, help='path to dir with train dataset chunks') parser.add_argument('-test', '--dataset_test', required=True, help='path to dir with test dataset chunks') parser.add_argument('-e', '--epochs', type=int, default=3, help='epochs count') parser.add_argument('-b', '--batch_size', type=int, default=32,
import os from logging import getLogger from tatau_core import settings from tatau_core.node.producer import Producer from tatau_core.utils.logging import configure_logging configure_logging('producer') logger = getLogger('tatau_core') def load_credentials(account_address_var_name): address = os.getenv(account_address_var_name) if address is None: raise ValueError('{} is not specified'.format(account_address_var_name)) storage_path = settings.KEYS_PATH dir_name = address.replace('0x', '') with open(os.path.join(storage_path, dir_name, 'rsa_pk.pem'), 'r') as f: pk = f.read() return address, pk.encode() def load_producer(): account_address, rsa_pk = load_credentials( account_address_var_name='PRODUCER_ACCOUNT_ADDRESS' ) p = Producer(