diff --git a/bots/markpages.py b/bots/markpages.py index 5dcf527..e2f587f 100644 --- a/bots/markpages.py +++ b/bots/markpages.py @@ -145,14 +145,10 @@ class MarkPagesBot( CurrentPageBot ): # sets 'current_page' on each treat() for redfam in self.redfams: # We need the talkpage (and only this) of each existing page - for talkpage in pagegenerators.PageWithTalkPageGenerator( - redfam.article_generator( - filter_existing=True, - exclude_article_status=["marked"] ), - return_talk_only=True ): - - # Add reference to redfam to talkpages - talkpage.redfam = redfam + for talkpage in redfam.article_generator( + filter_existing=True, + exclude_article_status=["marked"], + talkpages=True ): yield talkpage