def set_version_by_category(self, book_name):
book_ref = Ref(book_name)
if book_ref.is_tanach():
self.version_map[book_name] = 'Tanach with Text Only'
elif book_ref.is_bavli():
self.version_map[book_name] = 'Wikisource Talmud Bavli'
else:
raise AttributeError(
'{} does not match a default category, use set_version_by_book instead'
.format(book_name))
