import pandas as pd

print("################################")
from datasets import load_dataset
test_dset = load_dataset("ai-shift/compe_test", data_files={"test": "test.csv"})
print(test_dset)
print("################################")

df = pd.DataFrame([(f"testid{i:04}", 0) for i in range(837)], columns=["id", "pred"])
df.to_csv("submission.csv", index=None)