如何检测Databricks环境编程

steven35:

我正在写一个火花的工作需要可运本地以及对Databricks。

该代码必须在每种环境中(文件路径)略有不同,所以我试图找到一种方式,如果工作在Databricks运行检测。我发现,最好的办法,到目前为止是寻找在根目录一“DBFS”目录,如果它的存在,那么假设它是在Databricks运行。这并不觉得自己是正确的解决方案。有没有人有一个更好的主意吗?

pathikrit:

你可以简单地检查一个环境变量如存在:

def isRunningInDatabricks(): Boolean = 
  sys.env.contains("DATABRICKS_RUNTIME_VERSION")

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章