Running bert summarizer (run_summarization.py) gives the following error
Traceback (most recent call last):
File "run_summarization.py", line 15, in <module>
from .utils_summarization import (
ModuleNotFoundError: No module named '__main__.utils_summarization'; '__main__' is not a package
I've managed to fix this issue personally by changing "from .utils_summarization import" to "from utils_summarization import", though I don't know if this is due to a convention change in python module imports.
The problem arises when using:
Running the following command yields the error
python3 run_summarization.py --documents_dir ".../bertabs/dataset/input" --summaries_output_dir ".../bertabs/dataset/output" --no_cuda false --batch_size 4 --min_length 50 --max_length 200 --beam_size 5 --alpha 0.95 --block_trigram true
Steps to reproduce the behavior:
Traceback (most recent call last):
File "run_summarization.py", line 15, in <module>
from .utils_summarization import (
ModuleNotFoundError: No module named '__main__.utils_summarization'; '__main__' is not a package
It should import the module utils_summarization
transformers version: 2.8.0HELLO eeegnu, I'm also facing the same issue.

Environment Info :
Platform: Windows 10 64bit
Python version: 3.6.10
PyTorch version (GPU?): 1.4.0
Tensorflow version (GPU?): not installed (NA)
Hi!
Change: from .utils_summarization import (
CNNDMDataset,
build_mask,
compute_token_type_ids,
encode_for_summarization,
truncate_or_pad,
)
To: from utils_summarization import (
CNNDMDataset,
build_mask,
compute_token_type_ids,
encode_for_summarization,
truncate_or_pad,
)
This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.
Most helpful comment
Hi!
Change:
from .utils_summarization import ( CNNDMDataset, build_mask, compute_token_type_ids, encode_for_summarization, truncate_or_pad, )To:
from utils_summarization import ( CNNDMDataset, build_mask, compute_token_type_ids, encode_for_summarization, truncate_or_pad, )