BUG: sort all data prior to running model

Author: eigenfoo

bayesalpha/author_model.py

@@ -40,10 +40,7 @@ def __init__(self, data):
         self.num_backtests = data.meta_code_id.nunique()
 
         # Which algos correspond to which authors?
-        df = (data.sort_values(['meta_user_id',
-                                'meta_algorithm_id',
-                                'meta_code_id'])
-              .loc[:, ['meta_user_id', 'meta_algorithm_id']]
+        df = (data.loc[:, ['meta_user_id', 'meta_algorithm_id']]
               .drop_duplicates(subset='meta_algorithm_id', keep='first')
               .reset_index()
               .meta_user_id
@@ -200,7 +197,12 @@ def fit_authors(data,
     if sampler_type not in {'mcmc', 'vi'}:
         raise ValueError("sampler_type not in {'mcmc', 'vi'}")
 
+    # Check data and sort
     _check_data(data)
+    data = (data.sort_values(['meta_user_id',
+                              'meta_algorithm_id',
+                              'meta_code_id'])
+                .reset_index(drop=True))
 
     if seed is None:
         seed = int(random.getrandbits(31))