#
# Copyright (c) 2020 dividiti.
# See CK COPYRIGHT.txt for copyright details.
#
# SPDX-License-Identifier: BSD-3-Clause.
# See CK LICENSE.txt for licensing details.
#  

import os


def ck_preprocess(i):

  def dep_env(dep, var): return i['deps'][dep]['dict']['env'].get(var)

  inferencepath = dep_env('mlperf-inference', 'CK_ENV_MLPERF_INFERENCE')

  try:
    pythonpath = os.environ['PYTHONPATH'] + ":"
  except KeyError:
    pythonpath = ""

  os.environ['PYTHONPATH'] = pythonpath + \
                               os.path.join(inferencepath,"v0.7/speech_recognition/rnnt") + ":" + \
                               os.path.join(inferencepath,"v0.7/speech_recognition/rnnt/pytorch")

  return {'return': 0}