jolly-florist-66728
09/10/2024, 9:27 AM__init__.py file in your project’s folder.
Additionally, make sure that the container image is correctly specified in the @task decorator using the container_image parameter. Here is an example of how to specify a custom container image in a Flyte task:
from flytekit import task, workflow
@task(container_image='my-repo/my-image:latest')
def my_task():
import my_custom_library
print("Task executed with custom library!")
@workflow
def my_workflow():
my_task()
If the Docker image is not publicly available, ensure that you have configured the necessary credentials for pulling private images.
Would you like more detailed steps or further assistance?
Sources:
- Flyte Troubleshooting Guide
- Multiple Images in a Workflow
Slack Link
https://flyte-org.slack.com/archives/CP2HDHKE1/p1725960427.815959 👍 0 button 👎 0 button 💡 Hint: Mention @runllm in the thread for followups.