From 9dee8b153de1888e5794c9688ccdc442e46b0d8d Mon Sep 17 00:00:00 2001
From: Alberto Cattaneo <albertoc@graphcore.ai>
Date: Wed, 25 Oct 2023 10:33:59 +0000
Subject: [PATCH] fix typo

---
 besskge/pipeline.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/besskge/pipeline.py b/besskge/pipeline.py
index d77b330..73e6832 100644
--- a/besskge/pipeline.py
+++ b/besskge/pipeline.py
@@ -102,7 +102,8 @@ def __init__(
         inf_options.replication_factor = self.bess_module.sharding.n_shard
         inf_options.deviceIterations(self.batch_sampler.batches_per_step)
         inf_options.outputMode(poptorch.OutputMode.All)
-        inf_options.useIpuModel(use_ipu_model)
+        if use_ipu_model:
+            inf_options.useIpuModel(True)
         self.dl = self.batch_sampler.get_dataloader(options=inf_options, shuffle=False)
 
         self.poptorch_module = poptorch.inferenceModel(