Merge pull request #13 from davidgardenier/12-fix-nan-conversion

Fix NaN conversion and versioning on Github
TRASAL · Dec 9, 2022 · 988a5c5 · 988a5c5
2 parents f041e7b + b913a44
commit 988a5c5
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,