From 840e56e24de300cc1883dca2260a48291d45e1b2 Mon Sep 17 00:00:00 2001
From: John <john@zither-labs.com>
Date: Tue, 9 Jul 2024 21:12:35 +0200
Subject: [PATCH] cleaned multi_thermostat_driver and setup.py

---
 scripts/multi_thermostat_driver.py | 7 ++++---
 setup.py                           | 1 +
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/scripts/multi_thermostat_driver.py b/scripts/multi_thermostat_driver.py
index 4c7b9296..e2c643f3 100644
--- a/scripts/multi_thermostat_driver.py
+++ b/scripts/multi_thermostat_driver.py
@@ -38,7 +38,8 @@
 # Save cached weather data files? (Default: False)
 SAVE_CACHE = False
 CACHE_PATH = None  # Replace with location to save the weather cache files (e.g. '/tmp/epathermosat')
-TAU_SAVE_PATH = Path('/home/theurerjohn3/Documents/energystar/epa5/datadir/attempt1')
+TAU_SAVE_PATH = Path('outputs')
+TAU_SAVE_PATH.mkdir(parents=True, exist_ok=True)
 
 # This section finds the metadata files and data files for the thermostats.
 # These point to examples of the various styles of files
@@ -46,8 +47,8 @@
 # data in the same file.
 
 # Single Stage
-DATA_DIR = Path('../../datadir/EPA_Tau')
-METADATA_FILENAME = DATA_DIR / '2019_epa_tau.csv'
+DATA_DIR = Path(__file__).resolve().parents[1] / 'tests' / 'data' / 'single_stage'
+METADATA_FILENAME = DATA_DIR / 'metadata.csv'
 
 # Two Stage
 # DATA_DIR = os.path.join('..', 'tests', 'data', 'two_stage')
diff --git a/setup.py b/setup.py
index 901c57aa..e0894cd3 100644
--- a/setup.py
+++ b/setup.py
@@ -28,5 +28,6 @@
         'sqlalchemy<=1.4.46',
         'zipcodes==1.2.0',
     	'matplotlib==3.4.2',
+        'loguru',
         ],
 )