fix command job environment to support multiple data type (#41055)

Author: pdhotems

sdk/ml/azure-ai-ml/azure/ai/ml/_schema/job/parameterized_command.py

@@ -31,7 +31,7 @@ class ParameterizedCommandSchema(PathAwareSchema):
     environment = EnvironmentField(required=True)
     environment_variables = UnionField(
         [
-            fields.Dict(keys=fields.Str(), values=fields.Str()),
+            fields.Dict(),
             # Used for binding environment variables
             NestedField(InputLiteralValueSchema),
         ]

sdk/ml/azure-ai-ml/tests/command_job/unittests/test_command_job_schema.py

@@ -87,7 +87,7 @@ def test_deserialize_inputs(self):
 
     def test_deserialize_inputs_dataset(self):
         test_path = "./tests/test_configs/command_job/command_job_inputs_dataset_test.yml"
-        with open("./tests/test_configs/command_job/command_job_inputs_dataset_test.yml", "r") as f:
+        with open(test_path, "r") as f:
             target = yaml.safe_load(f)
         with open(test_path, "r") as f:
             cfg = yaml.safe_load(f)
@@ -96,6 +96,7 @@ def test_deserialize_inputs_dataset(self):
             internal_representation: CommandJob = CommandJob(**schema.load(cfg))
         source = internal_representation._to_rest_object()
         assert source.properties.inputs["test1"].uri == target["inputs"]["test1"]["path"]
+        assert source.properties.environment_variables == target["environment_variables"]
 
     def test_deserialize_inputs_dataset_short_form(self):
         test_path = "./tests/test_configs/command_job/command_job_inputs_dataset_short_form_test.yml"

sdk/ml/azure-ai-ml/tests/test_configs/command_job/command_job_inputs_dataset_test.yml

@@ -4,6 +4,11 @@ environment: azureml:AzureML-sklearn-1.0-ubuntu20.04-py38-cpu:33
 name: "test1"
 compute: "azureml:testCompute"
 experiment_name: mfe-test1
+environment_variables:
+  env-str: test_value
+  env-int: 1
+  env-bool: true
+  env-float: 1.0
 properties:
   test_property: test_value
 inputs: