fix tox environment

Author: rakeshcusat

sdks/python/apache_beam/yaml/json_utils.py

@@ -25,8 +25,6 @@
 from typing import Any
 from typing import Optional
 
-import jsonschema
-
 import apache_beam as beam
 from apache_beam.portability.api import schema_pb2
 from apache_beam.typehints import schemas
@@ -210,6 +208,7 @@ def json_parser(
   if json_schema is None:
     validate_fn = None
   else:
+    import jsonschema
     cls = jsonschema.validators.validator_for(json_schema)
     cls.check_schema(json_schema)
     validate_fn = _PicklableFromConstructor(

sdks/python/tox.ini

@@ -68,6 +68,7 @@ commands_post =
 commands = false {envname} is misconfigured
 
 [testenv:py{39,310,311,312,313}]
+extras = interactive,yaml
 commands_pre =
   python --version
   pip --version
@@ -85,6 +86,7 @@ commands_pre =
   pip --version
   # pip check
   bash {toxinidir}/scripts/run_tox_cleanup.sh
+extras = interactive,yaml
 commands =
   python apache_beam/examples/complete/autocomplete_test.py
   bash {toxinidir}/scripts/run_pytest.sh {envname} "{posargs}"