diff --git a/catalog_tools/io/client.py b/catalog_tools/io/client.py index 23ef3ec..bd07885 100644 --- a/catalog_tools/io/client.py +++ b/catalog_tools/io/client.py @@ -85,9 +85,12 @@ def get_events(self, start_time: Optional[datetime] = None, parser.parse(r.raw) df = Catalog.from_dict(catalog) - # if not include_uncertainty: - # wrong way round - # rgx = "(_uncertainty|_lowerUncertainty|" \ - # "_upperUncertainty|_confidenceLevel)$" - # df = df.filter(regex=rgx) + + if not include_uncertainty: + rgx = "(_uncertainty|_lowerUncertainty|" \ + "_upperUncertainty|_confidenceLevel)$" + # df = df.filter(regex=rgx) + cols = df.filter(regex=rgx).columns + df = df.drop(columns=cols) + return df