updating readme [skip ci]

ersilia-os · Mar 11, 2024 · 69e8464 · 69e8464
1 parent 8fd212e
commit 69e8464
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 17 deletions.
diff --git a/.gitattributes b/.gitattributes
@@ -1,7 +1 @@
-*.csv filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
 *.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.tsv filter=lfs diff=lfs merge=lfs -text
-*.pkl.gz filter=lfs diff=lfs merge=lfs -text
diff --git a/model/checkpoints/checkpoints.joblib b/model/checkpoints/checkpoints.joblib
diff --git a/model/framework/input.csv b/model/framework/input.csv
diff --git a/src/service.py b/src/service.py
@@ -18,6 +18,20 @@ def load_model(framework_dir, checkpoints_dir):
     mdl.load(framework_dir, checkpoints_dir)
     return mdl
 
+def String(x):
+    x = str(x)
+    if not x:
+        return None
+    if x == "nan":
+        return None
+    if x == "null":
+        return None
+    if x == "False":
+        return None
+    if x == "None":
+        return None
+    return x
+
 
 class Model(object):
     def __init__(self):
@@ -64,12 +78,8 @@ def run(self, smiles_list):
             reader = csv.reader(f)
             h = next(reader)
             R = []
-            samples = []
             for r in reader:
-                for x in r:
-                    sample = str(x)
-                    samples.append(sample)
-                R += [{"outcome": samples}] 
+                R += [{"outcome": [String(x) for x in r]}] 
 
         meta = {
             "outcome": h