FIx for GetAllVintageSeriesResult.to_pd_data_frame (#67)

macrobond · Aug 16, 2024 · 9c2ae57 · 9c2ae57
1 parent bde34f6
commit 9c2ae57
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/macrobond_data_api/common/types/get_all_vintage_series_result.py b/macrobond_data_api/common/types/get_all_vintage_series_result.py
@@ -1,4 +1,5 @@
 from dataclasses import dataclass
+from itertools import chain
 
 from typing import TYPE_CHECKING, Any, Dict, Sequence, overload, List
 
@@ -36,9 +37,12 @@ def to_pd_data_frame(self) -> "DataFrame":
         """
         import pandas  # pylint: disable=import-outside-toplevel
 
+        dates = list(set(chain.from_iterable((x.dates for x in self.series))))
+        dates.sort()
+
         df = pandas.DataFrame(
             {
-                **{"date": self.series[len(self.series) - 1].dates},
+                **{"date": dates},
             }
         )