Bump to spark 2.4.5 + minor improvements

This commit is contained in:
romainx
2020-02-11 21:30:47 +01:00
parent 3deefc7d16
commit 45d51e3b42
6 changed files with 52 additions and 9 deletions

View File

@@ -1,6 +1,7 @@
# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
import os
import logging
import docker
import pytest
@@ -10,6 +11,8 @@ from requests.packages.urllib3.util.retry import Retry
from requests.adapters import HTTPAdapter
LOGGER = logging.getLogger(__name__)
@pytest.fixture(scope='session')
def http_client():
"""Requests session with retries and backoff."""
@@ -72,9 +75,10 @@ class TrackedContainer(object):
all_kwargs = {}
all_kwargs.update(self.kwargs)
all_kwargs.update(kwargs)
LOGGER.info(f"Running {self.image_name} with args {all_kwargs} ...")
self.container = self.docker_client.containers.run(self.image_name, **all_kwargs)
return self.container
def remove(self):
"""Kills and removes the tracked docker container."""
if self.container: