Skip to content

Commit 7875e4b

Browse files
authored
Fix torchprofile import (#269)
1 parent cbbb2cb commit 7875e4b

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

docs/tutorials/pytorch/question-answering/bert-base-uncased_distilled-squad.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
},
3333
{
3434
"cell_type": "markdown",
35+
"id": "0b9ed365",
3536
"metadata": {},
3637
"source": []
3738
},

nlp_toolkit/optimization/dynamic/evolution.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@
2626
import timeit
2727
import numpy as np
2828
import torch
29-
import torchprofile
30-
29+
from nlp_toolkit.optimization.utils.utility import LazyImport
30+
torchprofile = LazyImport("torchprofile")
3131
logger = logging.getLogger(__name__)
3232

3333

0 commit comments

Comments
 (0)