control_script_cluster.py

import pandas as pd
from  run_tn_sim import run_tn_sim
import time
import argparse
import os

def process_task(args):
    topology, size, cycle, error = args
    if not os.path.exists(f"./errors_meas_actualfix_crosstalk/output_c{cycle}_a{topology}_s{size}_e{error}.csv"):
        vnes, tmi, shuttles = run_tn_sim(2000, error, topology, f"./compiled_circuits/c{cycle}_a{topology}_s{size}.qasm")
        
        data = {"vnes": vnes, "tmi": tmi, "shuttles":shuttles}
        dframe = pd.DataFrame(data)
        dframe.to_csv(f"./errors_meas_actualfix_crosstalk/output_c{cycle}_a{topology}_s{size}_e{error}.csv")
    # print("done")

if __name__ == '__main__':  
    
    parser = argparse.ArgumentParser()
    
    # Add arguments for the task ranges
    parser.add_argument('--topology', type=int, choices=range(1, 4), help="Error value")
    parser.add_argument('--size', type=int,choices=[4, 5, 6, 7], help="Error value ")
    parser.add_argument('--cycle', type=int, choices=range(1, 11), help="Error value ")
    parser.add_argument('--error', type=float, choices=[0.01, 0.03, 0.05 , 0.08, 0.1], help="Error value")

    # Number of worker processes to use    
    # Parse the arguments
    args = parser.parse_args()
    topology, size, cycle, error = vars(args).values()
    
    start_time = time.time()
    process_task((topology, size, cycle, error))
    end_time = time.time()
    
    print(f"Time taken for specific block: {end_time - start_time:.2f} seconds")