def tracking_uri_mock(tmpdir): try: mlflow.set_tracking_uri( path_to_local_sqlite_uri(os.path.join(tmpdir.strpath, 'mlruns'))) yield tmpdir finally: mlflow.set_tracking_uri(None)
def tracking_uri_mock(tmpdir, request): try: if "notrackingurimock" not in request.keywords: tracking_uri = path_to_local_sqlite_uri(os.path.join(tmpdir.strpath, "mlruns")) mlflow.set_tracking_uri(tracking_uri) os.environ["MLFLOW_TRACKING_URI"] = tracking_uri yield tmpdir finally: mlflow.set_tracking_uri(None) if "notrackingurimock" not in request.keywords: del os.environ["MLFLOW_TRACKING_URI"]
def _get_local_uri_or_none(uri): if uri == "databricks": return None, None parsed_uri = urllib.parse.urlparse(uri) if not parsed_uri.netloc and parsed_uri.scheme in ("", "file", "sqlite"): path = urllib.request.url2pathname(parsed_uri.path) if parsed_uri.scheme == "sqlite": uri = path_to_local_sqlite_uri(_MLFLOW_DOCKER_TRACKING_DIR_PATH) else: uri = path_to_local_file_uri(_MLFLOW_DOCKER_TRACKING_DIR_PATH) return path, uri else: return None, None