diff --git a/requirements.txt b/requirements.txt index 09a4fc1d..3212f694 100644 --- a/requirements.txt +++ b/requirements.txt @@ -27,4 +27,7 @@ mypy-protobuf>=3.0.0 pytest>=8.1.1 termcolor>=2.4.0 pandas>=2.2.2 -fastapi>=0.110.1 \ No newline at end of file +fastapi>=0.110.1 +termcolor +setuptools + diff --git a/swarms/splash_message.py b/swarms/splash_message.py index d12af1f6..0d076dd3 100644 --- a/swarms/splash_message.py +++ b/swarms/splash_message.py @@ -21,11 +21,7 @@ def splash(): if env_file_empty or config_display_message == "keep": print(""" -Warning: Some packages may not have been installed correctly. to ensure all packages are installed, run: - pip install python-dotenv - pip install termcolor - pip install setuptools -Pytorch will install with a default cpu config that may be different then the one for your machine. if this causes errors, run: +Warning: Pytorch will install with a default cpu config that may be different then the one for your machine. if this causes errors, run: pip uninstall pytorch pip install pytorch This message will only display the first time swarms is imported by default. @@ -35,7 +31,7 @@ Hit any key to continue, type "keep" | "yes" | "y" to keep showing this message, continue_or_not = input("keep showing message? ") - if (env_file_empty and continue_or_not != "keep") or (config_display_message == "keep" and continue_or_not == "dismiss"): + if (env_file_empty and continue_or_not not in ["yes", "y", "keep"]) or (config_display_message == "keep" and continue_or_not == "dismiss"): config["display_message"] = "dismiss" elif continue_or_not == "keep" or continue_or_not == None: config["display_message"] = "keep"