How do I disable gravity for a given object using Python API?

I created a box, and I don’t want it to fall under gravity but just keep floating in its position

    box = DynamicCuboid(prim_path="/World/box",
        position=np.array([0, 0, 1.5]),
        scale=np.array([.1, .1, .2]),
        color=np.array([.2, .3, 0.]))
    world.scene.add(box)

You could do something like this:

from pxr import PhysxSchema

box_prim = stage.GetPrimAtPath("/World/box")
physx_api = PhysxSchema.PhysxRigidBodyAPI.Apply(box_prim)
physx_api.CreateDisableGravityAttr(True)

Regards
Ales