make sure return df

e229caa7 · Dr.李 · ecfe86a2 · e229caa7
Commit e229caa7 authored Dec 08, 2018 by Dr.李
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

sqlengine.py alphamind/data/engines/sqlengine.py +1 -1

No files found.
--- a/alphamind/data/engines/sqlengine.py
+++ b/alphamind/data/engines/sqlengine.py
@@ -360,7 +360,7 @@ class SqlEngine(object):

        res['chgPct'] = df.chgPct
        res['secShortName'] = df['secShortName']
-        res = res.loc[ref_date]
+        res = res.loc[ref_date:ref_date, :]
        res.index = list(range(len(res)))
        return res