diff options
| -rw-r--r-- | ydb/tests/oss/ydb_sdk_import/__init__.py | 7 | ||||
| -rw-r--r-- | ydb/tests/oss/ydb_sdk_import/ya.make | 8 |
2 files changed, 13 insertions, 2 deletions
diff --git a/ydb/tests/oss/ydb_sdk_import/__init__.py b/ydb/tests/oss/ydb_sdk_import/__init__.py index 9a5f412784a..31ff3cca1f2 100644 --- a/ydb/tests/oss/ydb_sdk_import/__init__.py +++ b/ydb/tests/oss/ydb_sdk_import/__init__.py @@ -1,6 +1,11 @@ from ydb.tests.oss.canonical import is_oss +import os + if is_oss: - from ydb.public.sdk.python import ydb # noqa + if os.getenv('PYTHON2_YDB_IMPORT'): + from ydb.public.sdk.python2 import ydb # noqa + else: + from ydb.public.sdk.python3 import ydb # noqa else: import ydb # noqa diff --git a/ydb/tests/oss/ydb_sdk_import/ya.make b/ydb/tests/oss/ydb_sdk_import/ya.make index b7eedf44dec..24cc17d79e8 100644 --- a/ydb/tests/oss/ydb_sdk_import/ya.make +++ b/ydb/tests/oss/ydb_sdk_import/ya.make @@ -4,8 +4,14 @@ PY_SRCS( __init__.py ) +IF (PYTHON2) + ENV(PYTHON2_YDB_IMPORT='yes') + PEERDIR(ydb/public/sdk/python2) +ELSE() + PEERDIR(ydb/public/sdk/python3) +ENDIF() + PEERDIR( - ydb/public/sdk/python ydb/tests/oss/canonical ) |