def _py2j()

in sagemaker-pyspark-sdk/src/sagemaker_pyspark/wrapper.py [0:0]

• 13 lines of code
• 5 McCabe index (conditional complexity)

    def _py2j(self, arg):
        if isinstance(arg, dict):
            return ScalaMap(arg)._to_java()
        elif isinstance(arg, list):
            return ScalaList(arg)._to_java()
        elif isinstance(arg, StructType):
            return JavaWrapper._new_java_obj(
                "org.apache.spark.sql.types.DataType.fromJson", json.dumps(arg.jsonValue())
            )
        elif isinstance(arg, SageMakerJavaWrapper):
            return arg._to_java()
        else:
            return arg