Skip to content

Commit

Permalink
Merge branch 'master' into margo/readjson
Browse files Browse the repository at this point in the history
  • Loading branch information
AC-Dap authored Oct 8, 2023
2 parents 4ee6c29 + 82112c5 commit 31a21f7
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 0 deletions.
8 changes: 8 additions & 0 deletions ersilia/core/tracking.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import json
import pandas as pd

class RunTracker:
"""
Expand All @@ -8,13 +9,20 @@ class RunTracker:
NOTE: Currently, the Splunk connection is not set up. For now, we will print tracking results to the console.
"""

def read_csv(self, file):
# reads csv file and returns Pandas dataframe
return pd.read_csv(file)

def track(self, input, result, meta):
"""
Tracks the results after a model run.
"""

print("Run input file:", input)
print(self.read_csv(input))

print("Run output file:", result)
print(self.read_csv(result))

print("Model metadata:", meta)

Expand Down
4 changes: 4 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,10 @@ validators = [
{version="0.20.0", python="3.7.*"},
{version="~0.21.0", python=">=3.8"},
]
pandas = [
{version="1.2.4", python="3.7.*"},
{version="~1.3.0", python=">=3.8"},
]
h5py = "^3.7.0" # For compatibility with isaura
loguru = "^0.6.0" # For compatibility with isaura
pyairtable = "<2"
Expand Down

0 comments on commit 31a21f7

Please sign in to comment.