Python map_resource_to_metadata_typeの例

コード例 #1

ファイル: test_utils.py プロジェクト: wumingyu666/pipelines

    def test_map_resource_to_metadata_type_with_serializable_type(self):
        mb_sdk_type = List

        parameter_name, parameter_type = utils.map_resource_to_metadata_type(
            mb_sdk_type)
        self.assertEqual(parameter_name, 'exported_dataset')
        self.assertEqual(parameter_type, 'JsonArray')

コード例 #2

ファイル: test_utils.py プロジェクト: wumingyu666/pipelines

    def test_map_resource_to_metadata_type_with__Dataset_type(self):
        mb_sdk_type = '_Dataset'

        parameter_name, parameter_type = utils.map_resource_to_metadata_type(
            mb_sdk_type)
        self.assertEqual(parameter_name, 'dataset')
        self.assertEqual(parameter_type, 'Dataset')

コード例 #3

ファイル: test_utils.py プロジェクト: wumingyu666/pipelines

    def test_map_resource_to_metadata_type_with_mb_sdk_type(self):
        mb_sdk_type = aiplatform.Model

        parameter_name, parameter_type = utils.map_resource_to_metadata_type(
            mb_sdk_type)
        self.assertEqual(parameter_name, 'model')
        self.assertEqual(parameter_type, 'Model')