Skip to content

Commit 6e96cff

Browse files
authored
fix: fix submission file search and add TODO in env.py (microsoft#544)
1 parent fcf17d3 commit 6e96cff

File tree

2 files changed

+17
-1
lines changed

2 files changed

+17
-1
lines changed

rdagent/components/coder/data_science/workflow/eval_tests/submission_check.txt

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,24 @@
11
from pathlib import Path
22
import pandas as pd
33

4+
5+
"""
6+
find . | grep -i sample | grep -i submission | grep -v sample_submission.csv | grep -v zip_files | grep -v 'sample/'
7+
./denoising-dirty-documents/sampleSubmission.csv
8+
./the-icml-2013-whale-challenge-right-whale-redux/sampleSubmission.csv
9+
./text-normalization-challenge-russian-language/ru_sample_submission_2.csv.zip
10+
./text-normalization-challenge-russian-language/ru_sample_submission_2.csv
11+
./random-acts-of-pizza/sampleSubmission.csv
12+
./text-normalization-challenge-english-language/en_sample_submission_2.csv.zip
13+
./text-normalization-challenge-english-language/en_sample_submission_2.csv
14+
./detecting-insults-in-social-commentary/sample_submission_null.csv
15+
"""
16+
417
# Find sample submission file dynamically
518
input_dir = Path("/kaggle/input")
6-
sample_submission_files = list(input_dir.glob("*sample_submission*.csv"))
19+
# Look for common variations of sample submission filenames
20+
sample_submission_files = list(input_dir.glob("*sample_submission*.csv")) + \
21+
list(input_dir.glob("*sampleSubmission*.csv"))
722

823
if not sample_submission_files:
924
print("Error: No sample submission file found in /kaggle/input/")

rdagent/utils/env.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ class Env(Generic[ASpecificBaseModel]):
4646
"""
4747

4848
conf: ASpecificBaseModel # different env have different conf.
49+
# last_exit_code: # TODO: get the more concrete information about the exit code.
4950

5051
def __init__(self, conf: ASpecificBaseModel):
5152
self.conf = conf

0 commit comments

Comments
 (0)