Yasiel Puig’s bat is making the waves his Snapchat did last week. Now the question is whether the first-place Dodgers need the potential distraction.

The enigmatic outfielder has toned down his partying — or at least his social media documentation of it — and he’s been dominating Triple-A pitching.

In seven games with Oklahoma City, the 25-year-old is 13-for-27 (.481) with three homers and 11 RBIs. In 81 games with the Dodgers this season, Puig totaled just seven homers and 34 RBIs.

Though the Dodgers sent him down in part because of his cool bat, his reportedly divisive presence played a large role, too. Puig had rankled former manager Don Mattingly, and Dave Roberts immediately tried to make peace with the outfielder when he took over this season. It didn’t last.

“I think as a coach, we have to concern ourselves with the whole team,” Roberts said in late July, alluding to Puig, according to the Los Angeles Times. “As much as I care about him and I want him to be great, I mean, guys have to take ownership and be accountable for themselves, as a grown man.”

Without Puig’s presence, the Dodgers have quietly moved into first in the NL West, a half-game up on the Giants entering Wednesday’s play. In 13 games this month, the Dodgers are 7-6, but their scoring is up — from 4.5 runs per game to 5.2. That’s with right field being occupied by deadline acquisition Josh Reddick, who hasn’t provided the headache Puig did, or the results the Cuban did upon bursting onto the MLB scene. Reddick’s just 8-for-49 (.163) with no homers or RBIs in his 12 games.

A rejuvenated Puig could be a boost on the field to a team fighting for first. But a resurfaced Puig, causing the same havoc that got him banished to Oklahoma City, can derail a squad finally, and tenuously, atop its division.

“I think that’s up to Yasiel,” Roberts told the LA Times about a possible Puig callup. “If he chooses to continue to grow as a baseball player and as a man, then he’d be welcome back here.”