Unable to import 'foundry' on Code Workspaces (Jupyter)

In my Jupyter workspace, I’m running into 'ModuleNotFoundError: No module named ‘foundry’ on trying to use: from foundry.transforms import Dataset.

I didn’t change anything in the .yml file and I’ve already ‘added’ the dataset from the left pane. I created a new workspace and faced the same issue.

How could I fix this?