- Remove NaNs before converting to int in TNS

- Fix version "number" when using github branch instead of release
TRASAL · Dec 9, 2022 · b913a44 · b913a44
1 parent f041e7b
commit b913a44
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/frbcat/__version__.py b/frbcat/__version__.py
@@ -1 +1 @@
-version = 'github'
+__version__ = 'github'
diff --git a/frbcat/tns.py b/frbcat/tns.py
@@ -301,6 +301,9 @@ def clean_df(self, df):
                            'related_files': 'num_files',
                            'channels_no': 'num_channels'}, inplace=True)
 
+        # Some columns that will be converted to int can contain NaN, replace these by -1
+        df.fillna({'reports_id': -1, 'photometry_id': -1, 'num_channels': -1}, inplace=True)
+
         # Set dtypes
         df = df.astype({'tns_id': int,
                         'name': str,