Update criteo_dataset.py

[soma2000-lang]

Relevant Issue

#165

Checklist

• [x] I have followed the Contribution Guidelines and Code of Conduct
• [x] I have commented my code following the NimbleEdge Styleguide
• [x] I have labeled this PR with the relevant Type labels
• [x] My changes are covered by tests

[netlify[bot]]

✅ Deploy Preview for goofy-agnesi-4565cb canceled.

Name	Link
🔨 Latest commit	fcfe82c294a26e626723e39234914ac3b84cf53d
🔍 Latest deploy log	https://app.netlify.com/sites/goofy-agnesi-4565cb/deploys/627e0f874c29d600087c024d

[soma2000-lang]

@divyanshugit Please look into it.

[divyanshugit]

@soma2000-lang Instead of removing previous lines and adding lines. You can just add relevant things at relevant places.