fixed names

marty90 · Apr 2, 2020 · 7168a46 · 7168a46
1 parent 09fa936
commit 7168a46
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/profile_distribution.pickle → models.pickle b/profile_distribution.pickle → models.pickle
diff --git a/sample_from_distribution.py b/sample_from_distribution.py
@@ -1,22 +1,22 @@
 import pickle
 import sys
 
-DIST_FILE = "profile_distribution.pickle"
+DIST_FILE = "models.pickle"
 profiles = pickle.load(open(DIST_FILE, "rb"))
 
 # Parse Args
 _, operator, country, rat, quality = sys.argv
 
 # Test profile exists
 try:
-    _ = profiles.loc[(operator,country,rat, quality, "no_roaming")]
+    _ = profiles.loc[(operator,country,rat, quality)]
 except:
     print("error")
     exit(1)
 
 # Sample
 while True:
-    down, up, rtt = profiles.loc[(operator,country,rat, quality, "no_roaming")].kde.resample(1).transpose().squeeze()
+    down, up, rtt = profiles.loc[(operator,country,rat, quality)].kde.resample(1).transpose().squeeze()
     if down > 0 and up > 0 and rtt > 0:
         break