docformatter

Author: niklasmei

src/graphnet/models/pretraining_maskpred.py

@@ -1,4 +1,7 @@
-"""Exp. feature: Self-supervised pretraining using BERT-style mask prediction."""
+"""Experimental feature.
+
+Self-supervised pretraining using BERT-style mask prediction.
+"""
 
 from typing import Any, Tuple, Union, List, Type, Optional, Dict
 import os