diff options
| author | qrort <qrort@yandex-team.com> | 2023-03-17 11:51:02 +0300 |
|---|---|---|
| committer | qrort <qrort@yandex-team.com> | 2023-03-17 11:51:02 +0300 |
| commit | 0355d87069435442cdfe6f9ef56344128972df98 (patch) | |
| tree | ab10b07a21cf57f29c25c688d429bcbdd4771ef3 | |
| parent | 4f4d932dd1fea2a03e04f3838ecd406224dde81b (diff) | |
| download | ydb-0355d87069435442cdfe6f9ef56344128972df98.tar.gz | |
change oss import (because of rXXXXXX)
| -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 9a5f412784..31ff3cca1f 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 b7eedf44de..24cc17d79e 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 ) |