我正在构建一个Python包,数据科学家可以使用该包来管理他们的MLOps生命周期。现在,这个包可以在本地使用(通常在PyCharm上(,也可以在Databricks上使用。
我希望包的某些功能取决于它运行的位置,即如果它被Databricks笔记本调用,我希望它做一些不同的事情,如果它在本地运行,我希望完全做其他事情。
有什么方法可以确定它是从哪里打来的吗?
我有点怀疑我们是否可以使用下面的东西来检查你的代码是否在笔记本上运行,因为这将是一个将存储在你的Databricks环境中的包,
如何检查代码是否在IPython笔记本中执行?
我找到的解决方法是检查特定于databricks的环境变量。
import os
def is_running_in_databricks() -> bool:
return "DATABRICKS_RUNTIME_VERSION" in os.environ