def test_set_script_from_file(use_cluster): sent_script = read_script_from_file() c = Client(None, use_cluster) c.set_script_from_file( "test-script-file", osp.join(file_path, "./data_processing_script.txt") ) returned_script = c.get_script("test-script-file") assert sent_script == returned_script
def setup_resnet(model, device, batch_size, address, cluster=True): """Set and configure the PyTorch resnet50 model for inference :param model: path to serialized resnet model :type model: str :param device: CPU or GPU :type device: str :param batch_size: batch size for the Orchestrator (batches of batches) :type batch_size: int :param cluster: true if using a cluster orchestrator :type cluster: bool """ client = Client(address=address, cluster=cluster) client.set_model_from_file("resnet_model", model, "TORCH", device, batch_size) client.set_script_from_file("resnet_script", "../imagenet/data_processing_script.txt", device)
import os.path as osp from smartredis import Client # Construct a string holding the script file location file_path = osp.dirname(osp.abspath(__file__)) script_path = osp.join(file_path, "./data_processing_script.txt") # Connect to the Redis database db_address = "127.0.0.1:6379" client = Client(address=db_address, cluster=False) # Place the script in the database client.set_script_from_file( "test-script-file", osp.join(file_path, "./data_processing_script.txt") )
def test_bad_script_file(use_cluster): c = Client(None, use_cluster) with pytest.raises(FileNotFoundError): c.set_script_from_file("key", "not-a-file")