easy_rec/python/utils/compat.py

# -*- encoding:utf-8 -*-
# Copyright (c) Alibaba, Inc. and its affiliates.
# Date: 2019-10-12
# util to hanlde python2 python3 compatibility

import sys


def in_python2():
  return sys.version_info[0] == 2


def in_python3():
  return sys.version_info[0] == 3