Skip to content

Commit 5f4ce7e

Browse files
committed
remove type: ignore
1 parent dec1c9f commit 5f4ce7e

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/gfn/gflownet/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ def __init__(
223223
# Advisory: recurrent PF with non-recurrent PB is unusual
224224
# (tree DAGs typically prefer pb=None with constant_pb=True).
225225
# Import locally to avoid circular imports during module import time.
226-
from gfn.estimators import RecurrentDiscretePolicyEstimator # type: ignore
226+
from gfn.estimators import RecurrentDiscretePolicyEstimator
227227

228228
if isinstance(self.pf, RecurrentDiscretePolicyEstimator) and isinstance(
229229
self.pb, Estimator

src/gfn/gflownet/detailed_balance.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def __init__(
9595
)
9696

9797
# Disallow recurrent PF for transition-based DB
98-
from gfn.estimators import RecurrentDiscretePolicyEstimator # type: ignore
98+
from gfn.estimators import RecurrentDiscretePolicyEstimator
9999

100100
if isinstance(self.pf, RecurrentDiscretePolicyEstimator):
101101
raise TypeError(
@@ -220,7 +220,7 @@ def get_scores(
220220

221221
# Apply forward-looking if applicable
222222
if self.forward_looking:
223-
import warnings # type: ignore
223+
import warnings
224224

225225
warnings.warn(
226226
"Rewards should be defined over edges in forward-looking settings. "

0 commit comments

Comments
 (0)