chain

Author: Borda

.actions/assistant.py

@@ -289,7 +289,7 @@ def _bash_download_data(folder: str) -> List[str]:
             if ext not in AssistantCLI._EXT_ARCHIVE:
                 continue
             if ext in AssistantCLI._EXT_ARCHIVE_ZIP:
-                cmd += [f"mkdir -p {name}", f"unzip -o {fn} -d {name} {UNZIP_PROGRESS_BAR}"]
+                cmd += [f"unzip -o {fn} -d {name} {UNZIP_PROGRESS_BAR}"]
             else:
                 cmd += [f"tar -zxvf {fn} --overwrite"]
             cmd += [f"rm {fn}"]

templates/titanic/tutorial.py

@@ -94,12 +94,14 @@
     parameters=datamodule.parameters,
     batch_size=datamodule.batch_size,
 )
-predictions = trainer.predict(model, datamodule=dm, output="classes")
-print(predictions[0])
+preds = trainer.predict(model, datamodule=dm, output="classes")
+print(preds[0][:10])
 
 # %%
+import itertools  # noqa: E402]
 import numpy as np  # noqa: E402]
 
+predictions = list(itertools.chain(*preds))
 assert len(df_test) == len(predictions)
 
 df_test["Survived"] = np.argmax(predictions, axis=-1)