Merge pull request #165 from maread99/working

Fix 163
dpguthrie · Mar 18, 2023 · 1f131ed · 1f131ed
2 parents 444855e + e95865a
commit 1f131ed
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/yahooquery/utils/__init__.py b/yahooquery/utils/__init__.py
@@ -141,14 +141,14 @@ def _get_daily_index(data, index_utc, adj_timezone):
     times = index_local.time
 
     bv = times <= datetime.time(14)
-    if (bv).all():
-        index = index_local.floor("d")
+    if (bv).all() or data["meta"].get("exchangeName", "Nope") == "SAO":  # see issue 163
+        index = index_local.floor("d", ambiguous=True)
     elif (~bv).all():
-        index = index_local.ceil("d")
+        index = index_local.ceil("d", ambiguous=True)
     else:
         # mix of open times pre and post 14:00.
-        index1 = index_local[bv].floor("d")
-        index2 = index_local[~bv].ceil("d")
+        index1 = index_local[bv].floor("d", ambiguous=True)
+        index2 = index_local[~bv].ceil("d", ambiguous=True)
         index = index1.union(index2)
 
     index = pd.Index(index.date)