[BUG] final_preds_and_avg_probs.csv file is not generated after inference

[ujjwalbaid0408]

Describe the bug

A clear and concise description of what the bug is.

To Reproduce

Steps to reproduce the behavior:

1. Use the data BreastMNIST from https://medmnist.com/'
2. Run command 'python /content/GaNDLF/gandlf_run -c /content/config.yaml -i /content/medmnist/dataset/test_path.csv -m /content/model/ -t False -d cuda'

Expected behavior

After running inference for the classification task following .csv files must be generated:

• output_predictions.csv
• final_preds_and_avg_probs.csv
• logits.csv

But final_preds_and_avg_probs.csv and logits.csv are not generating.

Environment information

[sarthakpati]

Thanks for the report, @ujjwalbaid0408. What are the details about the environment, i.e., the result of python ./gandlf_debugInfo after activing the appropriate virtual environment?

[ujjwalbaid0408]

I am running this using Google Colab:

GANDLF version: 0.0.20-dev
Git hash: 50f369f
Platform: Linux-6.1.58+-x86_64-with-glibc2.35
Machine: x86_64
Processor: x86_64
Architecture: 64bit ELF
Python environment:
Version: 3.10.12
Implementation: CPython
Compiler: GCC 11.4.0
Build: main Nov 20 2023 15:14:05

[sarthakpati]

Thanks!

[Geeks-Sid]

@sarthakpati Assigning this to myself

[sarthakpati]

@ujjwalbaid0408 can you check if the fix works for you?

[ujjwalbaid0408]

This is working now as expected. Thank You!