Swift4画面をスリープさせない方法

この１行をappDelegateのapplicationに入れておけばスリープしません。

viewDidLoadに入れても反映されました。

UIApplication.shared.isIdleTimerDisabled = true

Note

実機実行をする際ケーブルに繋がっているとスリープ状態にならないようなので、確認をする際はケーブルを抜いた状態でスリープ状態にならないか確認するといいと思います。