I think you mean "from petsc4py import PETSc"