From 2f654032f0b4aa2147fd403d93828997c74db9ba Mon Sep 17 00:00:00 2001
From: Nicolas Schmid <nicolas@breiten.ch>
Date: Wed, 16 Aug 2023 12:59:13 +0200
Subject: [PATCH] feat: add option to keep or remove uncertainties when reading
 catalog

---
 catalog_tools/io/client.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

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