Skip to content

Commit

Permalink
fix: lint
Browse files Browse the repository at this point in the history
  • Loading branch information
soumik12345 committed Aug 19, 2024
1 parent 9180dde commit ace0fb5
Showing 1 changed file with 16 additions and 10 deletions.
26 changes: 16 additions & 10 deletions docs/notebooks/parse_arxiv_papers.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -110,9 +110,10 @@
},
"outputs": [],
"source": [
"from pydantic import BaseModel\n",
"from typing import List, Optional\n",
"\n",
"from pydantic import BaseModel\n",
"\n",
"\n",
"class Finding(BaseModel):\n",
" finding_name: str\n",
Expand All @@ -133,14 +134,20 @@
"\n",
"\n",
"class PaperInfo(BaseModel):\n",
" main_findings: List[Finding] # The main findings of the paper\n",
" main_findings: List[Finding] # The main findings of the paper\n",
" novel_methods: List[Method] # The novel methods proposed in the paper\n",
" existing_methods: List[Method] # The existing methods used in the paper\n",
" machine_learning_techniques: List[Method] # The machine learning techniques used in the paper\n",
" metrics: List[Evaluation] # The evaluation metrics used in the paper\n",
" github_repository: str # The link to the GitHub repository of the paper (if there is any)\n",
" hardware: str # The hardware or accelerator setup used in the paper\n",
" further_research: List[str] # The further research directions suggested in the paper"
" existing_methods: List[Method] # The existing methods used in the paper\n",
" machine_learning_techniques: List[\n",
" Method\n",
" ] # The machine learning techniques used in the paper\n",
" metrics: List[Evaluation] # The evaluation metrics used in the paper\n",
" github_repository: (\n",
" str # The link to the GitHub repository of the paper (if there is any)\n",
" )\n",
" hardware: str # The hardware or accelerator setup used in the paper\n",
" further_research: List[\n",
" str\n",
" ] # The further research directions suggested in the paper"
]
},
{
Expand Down Expand Up @@ -204,7 +211,6 @@
"import instructor\n",
"from openai import OpenAI\n",
"\n",
"\n",
"openai_client = OpenAI()\n",
"structured_client = instructor.from_openai(openai_client)"
]
Expand All @@ -224,11 +230,11 @@
"metadata": {},
"outputs": [],
"source": [
"import requests\n",
"from io import BytesIO\n",
"\n",
"import pymupdf\n",
"import pymupdf4llm\n",
"import requests\n",
"\n",
"\n",
"class ArxivModel(weave.Model):\n",
Expand Down

0 comments on commit ace0fb5

Please sign in to comment.