diff --git a/.idea/workspace.xml b/.idea/workspace.xml index a5ad9c4..78a0b14 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -214,7 +214,7 @@ - + diff --git a/serve.py b/serve.py index 2acd458..8c28a6e 100644 --- a/serve.py +++ b/serve.py @@ -212,7 +212,7 @@ class MyHandler(BaseHTTPRequestHandler): with open(config.train_data_path, 'r') as f: q = deque(f, config.time_step - 1) df = pd.read_csv(StringIO(''.join(q)), - header=['job', 'model', 'time', 'utilCPU', 'utilGPU', 'pre', 'main', 'post']) + usecols=['job', 'model', 'time', 'utilCPU', 'utilGPU', 'pre', 'main', 'post']) print(df) df = df.append(data, ignore_index=True)