Unable to use the statsD logging. Init-container fails.

[maulikjs]

Whenever I try and provide this config in the superset config file

from superset.stats_logger import StatsdStatsLogger
STATS_LOGGER = StatsdStatsLogger(host='localhost', port=8125, prefix='superset')

as per instructions here: https://superset.incubator.apache.org/installation.html#statsd-logging

The Init container fails with this log:

ERROR:root:Failed to import config for SUPERSET_CONFIG_PATH=/etc/superset/superset_config.py
Traceback (most recent call last):
  File "/opt/app-root/lib/python3.6/site-packages/superset/config.py", line 633, in <module>
    override_conf = imp.load_source("superset_config", cfg_path)
  File "/opt/app-root/lib64/python3.6/imp.py", line 172, in load_source
    module = _load(spec)
  File "<frozen importlib._bootstrap>", line 684, in _load
  File "<frozen importlib._bootstrap>", line 665, in _load_unlocked
  File "<frozen importlib._bootstrap_external>", line 678, in exec_module
  File "<frozen importlib._bootstrap>", line 219, in _call_with_frames_removed
  File "/etc/superset/superset_config.py", line 2, in <module>
    from superset.stats_logger import StatsdStatsLogger
ImportError: cannot import name 'StatsdStatsLogger'
Usage: flask fab create-admin [OPTIONS]

Error: While importing "superset", an ImportError was raised:

Traceback (most recent call last):
  File "/opt/app-root/lib/python3.6/site-packages/flask/cli.py", line 235, in locate_app
    __import__(module_name)
  File "/opt/app-root/lib/python3.6/site-packages/superset/__init__.py", line 35, in <module>
    from superset import config
  File "/opt/app-root/lib/python3.6/site-packages/superset/config.py", line 633, in <module>
    override_conf = imp.load_source("superset_config", cfg_path)
  File "/opt/app-root/lib64/python3.6/imp.py", line 172, in load_source
    module = _load(spec)
  File "/etc/superset/superset_config.py", line 2, in <module>
    from superset.stats_logger import StatsdStatsLogger
ImportError: cannot import name 'StatsdStatsLogger'

Have you gotten this working before? Any pointers will be helpful :)