search

lowe-lab-ucl / cellx-predict

Explainable AI model of cell behavior
BSD 3-Clause "New" or "Revised" License
20 stars 10 forks source link

List tfrecord files in log file #42

Closed quantumjot closed 1 year ago

quantumjot commented 1 year ago

https://github.com/lowe-lab-ucl/cellx-predict/blob/36fe74354fcb8251167a8d9c0fc9a18474a68141/cellxpredict/session.py#L9-L12

Update json_data dict with the list of tfrecord files in config.src_dir: https://github.com/lowe-lab-ucl/cellx-predict/blob/36fe74354fcb8251167a8d9c0fc9a18474a68141/cellxpredict/config.py#L18

quantumjot commented 1 year ago

Should just be able to do something like:

src_files = list(Path(config.src_dir).glob("*.tfrecord"))
json_data.update({"src_files": src_files})