from DataProcessing.FloorRW import FloorRW floor_rw = FloorRW() floor_rw.generate_floors()