cog predict does not download a file when I return its Path specifying its extension

[ericguizzo]

cog predict does not download a file if I explicitly write the extension in the name of the output Path (I am under Mac OS).

Example:

    Inside predict function:
    output_path = Path(tempfile.mkdtemp()) / "output.txt"
    with open(str(output_path), "w") as f:
        f.write("hello")
    return output_path 

This only prints the file's content on the terminal and does not download it.
But if I remove the ".txt" from the filename the download works, but the file is a .bin and it does not contain the text.

[bfirsh]

Ah thank you -- yes Cog does some fuzzy matching to try and figure out what file types. We've been thinking about how to have a more robust type system: #205

In this case, one way to get this as a file would be to use the -o option, or to pipe to a file. For example:

cog predict -i ... -o output.txt
cog predict -i ... > output.txt

[zeke]

@allcontributors please add @ericguizzo for bugs 🪲

[allcontributors]

@zeke

I've put up a pull request to add @ericguizzo! 🎉

[anish-sahoo]

Closing as the type-system redesign replaced fuzzy filename matching with explicit output types, so Path outputs are now correctly saved with appropriate MIME types and extensions.